1. 学ぶ
  2. /
  3. コース
  4. /
  5. PyTorch による効率的な AI モデルトレーニング

Connected

演習

モデルを評価モードに設定する

いよいよ言語モデルを評価モードに設定しましょう。推論時にモデルが評価モードになっていない場合、バッチ正規化やドロップアウトなどのレイヤーがモデルの動作に影響を与え、翻訳品質が不安定になることがあります。モデルを評価するためのループを構築しましょう。

model、eval_dataloader、accelerator、metric はあらかじめ読み込まれています。

指示

100 XP
  • データセットのバッチをループ処理する前に、モデルを評価モードに設定します。
  • Accelerator の .gather_for_metrics() メソッドを使って、デバイス間で predictions と labels を集約し、評価指標を計算できるようにします。
  • 最後に評価指標を計算します。