ćwiczenie

Konfiguracja optymalizatora

Skoro mamy już logikę trenowania, trzeba teraz określić, jak optymalizować parametry modelu.

W tym ćwiczeniu uzupełnisz metodę configure_optimizers w module PyTorch Lightning używanym do klasyfikacji obrazów. Twoim celem jest skonfigurowanie optymalizatora, który będzie aktualizować parametry modelu podczas trenowania. Użyj do tego optymalizatora Adam ze współczynnikiem uczenia 1e-3.

Instrukcje

100 XP

Utwórz optymalizator Adam, korzystając z parametrów modelu i ustawiając współczynnik uczenia na 1e-3.

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

Instrukcje

ćwiczenie