IniziaInizia gratis

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

Visualizza il corso

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,
)
Modifica ed esegui il codice