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
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)