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

Connected

演習

評価指標の記録

パフォーマンス指標を追跡することで、精度の低下をモニタリングし、モデルをいつ更新すべきかを判断できます。ここでは、評価ループが終了した後に指標を記録するように設定しましょう。

以下のデータが事前に読み込まれています。

  • accelerator は Accelerator のインスタンスです
  • eval_metric は accuracy や f1 などの指標を格納した辞書です
  • num_epochs はエポック数です

指示

100 XP
  • モデルの評価指標を記録するメソッドを呼び出します。
  • 評価指標として "accuracy" と "f1" スコアを記録します。
  • ループ変数 epoch を step パラメータに渡して、エポック番号を追跡します。