LoslegenKostenlos loslegen

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

Kurs anzeigen

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)
Code bearbeiten und ausführen