Sperimentare con il learning rate
In questo esercizio, il tuo obiettivo è trovare il learning rate ottimale in modo che l'ottimizzatore trovi il minimo della funzione non convessa \(x^{4} + x^{3} - 5x^{2}\) in dieci passaggi.
Sperimenterai con tre valori diversi di learning rate. Per questo problema, prova valori di learning rate tra 0.001 e 0.1.
Ti viene fornita la funzione optimize_and_plot() che prende il learning rate come primo argomento. Questa funzione eseguirà 10 passaggi 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 learning rate value
lr0 = ____
optimize_and_plot(lr=lr0)