IniziaInizia gratis

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

Visualizza il corso

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