Optimizatori
Este momentul să explorezi diferiții optimizatori pe care îi poți folosi pentru antrenarea modelului tău.
O funcție personalizată numită train_model(optimizer, net, num_epochs) a fost deja definită. Aceasta primește optimizatorul, modelul și numărul de epoci ca argumente, rulează buclele de antrenament și afișează pierderea la final.
Hai să folosim train_model() pentru câteva antrenamente scurte cu diferiți optimizatori și să comparăm rezultatele!
Acest exercițiu face parte din cursul
Deep Learning intermediar cu PyTorch
Exercițiu interactiv practic
Încearcă acest exercițiu completând acest cod de exemplu.
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,
)