Implementace validačního kroku
Po natrénování neuronové sítě je důležité sledovat její výkon v průběhu tréninku. Pomocí PyTorch Lightning implementuj metodu validation_step(), která bude při každé epoše vypočítávat a zaznamenávat validační ztrátu.
Toto cvičení je součástí kurzu
Škálovatelné modely AI s PyTorch Lightning
Pokyny k cvičení
- Vypočítej predikce modelu pro vstupní dávku dat.
- Vypočítej validační ztrátu pomocí
F.cross_entropy(). - Zaznamenej validační ztrátu pomocí
self.log()pod názvemval_loss.
Interaktivní cvičení na vyzkoušení si v praxi
Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.
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)