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
Vervollständige den Beispielcode, um diese Übung erfolgreich abzuschließen.
# Try a first value for momentum
mom0 = ____
optimize_and_plot(momentum=mom0)