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

演習

Trainer での AdamW の使用

言語翻訳を簡略化する Transformer モデルのトレーニングを始めましょう。最初のステップとして、ベースラインとなる AdamW オプティマイザを使用し、Trainer インターフェースで手軽にセットアップします。AdamW の optimizer を使用するように Trainer を設定してください。

AdamW は torch.optim からあらかじめインポートされています。また、model、training_args、train_dataset、validation_dataset、compute_metrics などのトレーニング用オブジェクトも事前に読み込まれています。

指示

100 XP
  • model のパラメータを AdamW の optimizer に渡しましょう。
  • optimizer を Trainer に渡しましょう。