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. Experimentiere mit zwei verschiedenen Momentum-Werten. Für diese Aufgabe 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. Diese Funktion führt 20 Schritte des SGD-Optimierers aus und zeigt die Ergebnisse an.
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)