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

演習

Accelerator を使ったトレーニングループの構築

言語翻訳サービス向けのトレーニングループを実装する準備が整いました。Accelerator が分散トレーニングのために PyTorch ループをどのように変更するかを確認したので、トレーニングループで Accelerator クラスを活用しましょう。

以下のデータはあらかじめ読み込まれています。

  • accelerator は Accelerator のインスタンスです
  • train_dataloader、optimizer、model、lr_scheduler は定義済みで、Accelerator によって準備されています

指示

100 XP
  • optimizer を呼び出して、勾配をゼロにリセットします。
  • モデルのパラメータを更新します。
  • optimizer の学習率を更新します。