НачатьНачать бесплатно

Реализация шага валидации

После обучения нейронной сети важно отслеживать её качество в процессе обучения. С помощью PyTorch Lightning реализуйте метод validation_step(), который будет вычислять и записывать значение потерь на валидационной выборке на каждой эпохе.

Это упражнение является частью курса

Масштабируемые модели ИИ с PyTorch Lightning

Посмотреть курс

Инструкции к упражнению

  • Вычислите предсказания модели для входного батча.
  • Рассчитайте потери на валидационной выборке с помощью F.cross_entropy().
  • Запишите значение потерь с помощью self.log() под именем val_loss.

Интерактивное практическое упражнение

Попробуйте выполнить это упражнение, дополнив этот пример кода.

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)
Редактировать и запускать код