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

演習

分散学習用にモデルを準備する

Hugging Face の Accelerator ライブラリを使って、言語翻訳モデルを学習させることにしました。いよいよ、モデルを分散学習用に準備しましょう。

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

  • accelerator は Accelerator のインスタンスです
  • model、optimizer、train_dataloader、lr_scheduler が定義済みです

指示

100 XP
  • 分散学習用にオブジェクトを準備するメソッドを呼び出しましょう。
  • 出力の順序に合わせて、学習に必要なオブジェクトを位置引数として accelerator.prepare() に渡しましょう。