Mengimplementasikan langkah validasi
Setelah kita melatih model neural network, kita perlu memantau kinerjanya selama pelatihan. Dengan PyTorch Lightning, implementasikan metode validation_step() untuk menghitung dan mencatat validation loss pada setiap epoch.
Latihan ini adalah bagian dari kursus
Model AI yang Dapat Diskalakan dengan PyTorch Lightning
Petunjuk latihan
- Hitung prediksi menggunakan model pada batch masukan.
- Hitung validation loss menggunakan
F.cross_entropy(). - Catat validation loss dengan
self.log()sebagaival_loss.
Latihan interaktif praktis
Cobalah latihan ini dengan menyelesaikan kode contoh berikut.
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)