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
<Kurs>Einführung in Deep Learning mit PyTorch</Kurs>Interaktive praktische Übung
Versuche dich an dieser Übung, indem du diesen Beispielcode vervollständigst.
# Try a first value for momentum
mom0 = ____
optimize_and_plot(momentum=mom0)