1. 学ぶ
    2. /
  2. コース
    3. /
  3. Hugging Face で学ぶマルチモーダルモデル
Connected

演習

CV のファインチューニング：Trainer の設定

データセットを用意し、事前学習モデルを新しいクラスに合わせて調整できたので、次は Trainer を設定します。

TrainingArguments と Trainer は transformers ライブラリから読み込まれています。モデル（model）とデータセット（dataset）は、前の手順で設定したものが読み込まれています。

指示

100 XP
  • 学習率を 6e-5 に調整します。
  • Trainer インスタンスにモデル、学習データ、テストデータを指定します。