Ottimizzatori
È il momento di esplorare i diversi ottimizzatori che puoi usare per addestrare il tuo modello.
È stata definita per te una funzione personalizzata chiamata train_model(optimizer, net, num_epochs). Prende in input l’ottimizzatore, il modello e il numero di epoche, esegue i cicli di addestramento e stampa la loss di training alla fine.
Usiamo train_model() per avviare alcuni brevi addestramenti con ottimizzatori diversi e confrontarne i risultati!
Questo esercizio fa parte del corso
Deep Learning intermedio con PyTorch
Esercizio pratico interattivo
Prova a risolvere questo esercizio completando il codice di esempio.
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,
)