övning

Implementacja kroku treningowego

W tym ćwiczeniu zaimplementujesz metodę training_step() w module PyTorch Lightning przeznaczonym do klasyfikacji obrazów. Twoja implementacja powinna rozpakowywać wsad obrazów i etykiet, obliczać predykcje modelu za pomocą przejścia w przód (forward pass), wyznaczać stratę entropii krzyżowej oraz rejestrować stratę treningową.

Instruktioner

100 XP

Upewnij się, że predykcje są obliczane przy użyciu przejścia w przód (forward pass).
Oblicz stratę entropii krzyżowej.
Zarejestruj stratę treningową.

.css-6su6fj{-webkit-flex-shrink:0;-ms-flex-negative:0;flex-shrink:0;}övning

Instruktioner

övning