Aan de slagGa gratis aan de slag

Experimenteren met de learning rate

In deze oefening is je doel om de optimale learning rate te vinden zodat de optimizer binnen tien stappen het minimum van de niet-convexe functie \(x^{4} + x^{3} - 5x^{2}\) kan vinden.

Je gaat experimenteren met drie verschillende waarden voor de learning rate. Probeer voor dit probleem waarden tussen 0,001 en 0,1.

Je krijgt de functie optimize_and_plot() die de learning rate als eerste argument gebruikt. Deze functie voert 10 stappen uit met de SGD-optimizer en toont de resultaten.

Deze oefening maakt deel uit van de cursus

Introductie tot Deep Learning met PyTorch

Cursus bekijken

Praktische interactieve oefening

Probeer deze oefening eens door deze voorbeeldcode in te vullen.

# Try a first learning rate value
lr0 = ____
optimize_and_plot(lr=lr0)
Code bewerken en uitvoeren