ćwiczenie

Implementacja kroku walidacji

Po wytrenowaniu modelu sieci neuronowej warto monitorować jego wydajność podczas treningu. Używając PyTorch Lightning, zaimplementuj metodę validation_step(), która oblicza i rejestruje stratę walidacyjną po każdej epoce.

Instrukcje

100 XP

Oblicz predykcje, uruchamiając model na wejściowej paczce danych.
Oblicz stratę walidacyjną za pomocą F.cross_entropy().
Zarejestruj stratę walidacyjną przy użyciu self.log() pod nazwą val_loss.

.css-6su6fj{-webkit-flex-shrink:0;-ms-flex-negative:0;flex-shrink:0;}ćwiczenie

Instrukcje

ćwiczenie