IniziaInizia gratis

Sperimentare con il momentum

In questo esercizio, il tuo obiettivo è trovare il momentum ottimale affinché l'ottimizzatore individui il minimo della seguente funzione non convessa \(x^{4} + x^{3} - 5x^{2}\) in 20 step. Sperimenterai con due valori diversi di momentum. Per questo problema, il learning rate è fissato a 0.01.

Ti viene fornita la funzione optimize_and_plot() che accetta in input il parametro momentum. Questa funzione eseguirà 20 step dell'ottimizzatore SGD e mostrerà i risultati.

Questo esercizio fa parte del corso

Introduzione al Deep Learning con PyTorch

Visualizza il corso

Esercizio pratico interattivo

Prova a risolvere questo esercizio completando il codice di esempio.

# Try a first value for momentum
mom0 = ____
optimize_and_plot(momentum=mom0)
Modifica ed esegui il codice