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

Connected

演習

Trainer のセットアップ

このモデルは、翻訳の読みやすさを向上させるために、複雑なフレーズをよりシンプルで分かりやすい言い換えに置き換えます。たとえば、「The inclement weather conditions precipitated the postponement of the outdoor event」という表現を「The bad weather caused the outdoor event to be delayed.」に簡略化できます。言語翻訳サービスの学習を準備するために Trainer を構築しましょう。trainer.train() の呼び出しを含むため、実行に少し時間がかかります。

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

  • model:Transformer モデル
  • dataset:文の言い換えデータセットである MRPC データセット
  • compute_metrics:正解率と F1 スコアを返す関数
  • training_args:前の演習で定義済み

指示

100 XP
  • model を Trainer() クラスに渡してください。
  • 学習の引数を Trainer() クラスに渡してください。
  • メトリクスを計算する関数を Trainer() クラスに渡してください。
  • trainer が選択したデバイスを出力してください。