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