Trainer のセットアップ
学習用の引数を用意できたら、微調整したモデルを使う前に Trainer を設定します。
TrainingArguments と Trainer、そしてこれまでに作成した model、tokenizer、training_args は読み込まれています。
注: この演習では処理を速くするため、実際の学習は無効化されています。
この演習はコースの一部です
Pythonで学ぶ LLM 入門
演習の手順
Trainer()オブジェクトをセットアップします。- 先ほど定義した学習用引数とトークナイザーを割り当てます。
- モデルを学習させます。
実践的なインタラクティブ演習
このサンプルコードを完成させて、この演習に挑戦してみましょう。
# Set up the trainer object
trainer = ____(
model=model,
# Assign the training arguments and tokenizer
____,
train_dataset=tokenized_training_data,
eval_dataset=tokenized_test_data,
____
)
# Train the model
____