Implementación del paso de validación
Una vez que hemos entrenado un modelo de red neuronal, necesitamos supervisar su rendimiento durante el entrenamiento. Utilizando PyTorch Lightning, implementa el método « validation_step()
» para calcular y registrar la pérdida de validación en cada época.
Este ejercicio forma parte del curso
Modelos de IA escalables con PyTorch Lightning
Instrucciones del ejercicio
- Calcula predicciones utilizando el modelo en el lote de entrada.
- Calcula la pérdida de validación utilizando
F.cross_entropy()
. - Registra la pérdida de validación con
self.log()
comoval_loss
.
Ejercicio interactivo práctico
Prueba este ejercicio completando el código de muestra.
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)