實作驗證步驟
當我們訓練好一個神經網路模型後，需要在訓練過程中監控其效能。請使用 PyTorch Lightning，實作 validation_step() 方法，在每個 epoch 計算並記錄驗證損失。
本練習屬於課程
Scalable AI Models with PyTorch Lightning
練習說明
- 使用模型對輸入批次產生預測。
- 使用
F.cross_entropy()計算驗證損失。 - 以
self.log()將驗證損失記錄為val_loss。
動手互動練習
試著完成這個範例程式碼，體驗一下這個練習。
import torch.nn.functional as F
def validation_step(self, batch, batch_idx):
x, y = batch
# Compute predictions using the model
preds = ____(x)
# Calculate validation loss
loss = F.____(preds, y)
# Log the validation loss
self.____('val_loss', loss)