Optymalizatory
Czas poznać różne optymalizatory, których możesz używać podczas trenowania modelu.
Zdefiniowano dla ciebie niestandardową funkcję train_model(optimizer, net, num_epochs). Przyjmuje ona optymalizator, model i liczbę epok jako argumenty, uruchamia pętle treningowe i wyświetla stratę treningową na końcu.
Użyjmy train_model(), aby przeprowadzić kilka krótkich treningów z różnymi optymalizatorami i porównać wyniki!
To ćwiczenie jest częścią kursu
Głębokie uczenie z PyTorch – poziom średnio zaawansowany
Interaktywne ćwiczenie praktyczne
Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.
import torch.optim as optim
net = Net()
# Define the SGD optimizer
optimizer = optim.____(net.parameters(), lr=0.001)
train_model(
optimizer=optimizer,
net=net,
num_epochs=10,
)