訓練與評估
在這個練習中，你會把目前學到的重點整合起來，針對埃塞俄比亞手寫 MNIST 字元的真實世界資料集進行神經網路的訓練與評估。
ImageClassifier 是使用 PyTorch Lightning 實作的預先定義神經網路模型。它包含用於特徵萃取的卷積層、引入非線性的啟動函式，以及用於分類的全連接層。
Ethiopic MNIST 資料集已為你預先匯入。
本練習屬於課程
Scalable AI Models with PyTorch Lightning
練習說明
- 匯入
Trainer。 - 定義
ImageClassifier模型與訓練器。 - 訓練模型。
- 在驗證集上評估模型。
動手互動練習
試著完成這個範例程式碼，體驗一下這個練習。
# Import the Trainer
from lightning.pytorch import ____
# Define ImageClassifier model & trainer and set epoch parameter
model = ____()
trainer = ____(max_epochs=5)
# Train the model
trainer.fit(____, train_loader, val_loader)
# Evaluate the model
val_results = trainer.____(____, val_loader)
print("Validation Accuracy:", val_results[0]["val_acc"])