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 merupakan bagian dari kursus
Model AI yang Dapat Diskalakan dengan PyTorch Lightning
Instruksi 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 langsung praktik
Cobalah latihan ini dengan melengkapi kode contoh ini.
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)