Aan de slagGa gratis aan de slag

Experimenteren met momentum

In deze oefening is je doel om het optimale momentum te vinden zodat de optimizer binnen 20 stappen het minimum kan vinden van de volgende niet-convexe functie \(x^{4} + x^{3} - 5x^{2}\). Je gaat experimenteren met twee verschillende momentumwaarden. Voor dit probleem staat de learning rate vast op 0,01.

Je krijgt de functie optimize_and_plot() die de parameter momentum als invoer accepteert. Deze functie voert 20 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 value for momentum
mom0 = ____
optimize_and_plot(momentum=mom0)
Code bewerken en uitvoeren