1. Učit se
    2. /
  2. Kurzy
    3. /
  3. PyTorch による効率的な AI モデルトレーニング
Connected

Cvičení

Accelerator による自動デバイス配置

大規模なデータセットで会話 AI モデルを学習させるために、モデルを GPU に移動することにしました。自動デバイス配置には Accelerator を活用します。この演習は実際には CPU 上で実行されますが、GPU で実行する場合もコードは同じです。

BERT ベースのモデルが model としてあらかじめ読み込まれています。

Pokyny

100 XP
  • 適切なクラスをインスタンス化して、accelerator オブジェクトを宣言しましょう。
  • accelerator オブジェクトを使って、GPU での分散学習用にモデルを準備しましょう。