Lightning Trainer の実行
モデルの定義が完了したら、次は Lightning の Trainer を設定して学習ループを実行しましょう。この演習では、モデル、データローダー、学習設定を組み合わせる方法を学びます。より高度な学習に進む前に、この設定方法を理解しておくことが重要です。
torch と、pl としてインポートされた lightning.pytorch はすでに読み込まれています。また、前の演習で定義したクラスを使って model も初期化済みです。
この演習はコースの一部です
PyTorch Lightning によるスケーラブルな AI モデルの構築
演習の手順
Trainerを3エポックで設定します。- 事前定義された
modelを使って学習を開始します。
実践的なインタラクティブ演習
このサンプルコードを完成させて、この演習に挑戦してみましょう。
from lightning.pytorch import Trainer
# Configure the Trainer with max epochs
trainer = ____(max_epochs=____, devices=1)
# Start training with the predefined model
trainer.____(____, train_dataloader, val_dataloader)