ComenzarEmpieza gratis

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

Ver curso

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() como val_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)
Editar y ejecutar código