Mit dem Momentum experimentieren
In dieser Übung ist es dein Ziel, das optimale Momentum zu finden, so dass der Optimierer das Minimum der folgenden nicht-konvexen Funktion \(x^{4} + x^{3} - 5x^{2}\) in 20 Schritten finden kann. Du wirst mit zwei verschiedenen Momentum-Werten experimentieren. Für dieses Problem wird die Lernrate auf 0,01 festgelegt.
Dir wird die Funktion optimize_and_plot()
zur Verfügung gestellt, die als Eingabe den Parameter momentum
akzeptiert. Mit dieser Funktion werden 20 Schritte des SGD-Optimierers ausgeführt und die Ergebnisse angezeigt.
Diese Übung ist Teil des Kurses
Einführung in Deep Learning mit PyTorch
Interaktive Übung
Versuche dich an dieser Übung, indem du diesen Beispielcode vervollständigst.
# Try a first value for momentum
mom0 = ____
optimize_and_plot(momentum=mom0)