執行 Lightning Trainer
現在你已經定義好模型，可以設定 Lightning 的 Trainer 來執行訓練迴圈。這個練習會帶你了解如何整合模型、資料載入器與訓練設定——熟悉這套流程後，才能順利邁向更進階的訓練。
torch 與 lightning.pytorch（匯入為 pl）已為你預先載入。model 也已經用你在上一個練習中定義的類別初始化完成。
本練習屬於課程
Scalable AI Models with PyTorch Lightning
練習說明
- 使用 3 個 epochs 設定
Trainer。 - 使用預先定義的
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)