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. 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

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