Mit der Lernrate experimentieren
In dieser Übung ist es dein Ziel, die optimale Lernrate zu finden, damit der Optimierer das Minimum der nicht-konvexen Funktion \(x^{4} + x^{3} - 5x^{2}\) in zehn Schritten finden kann.
Du wirst mit drei verschiedenen Werten als Lernraten experimentieren. Probiere für diese Aufgabe Lernratenwerte zwischen 0,001 und 0,1 aus.
Du erhältst die Funktion optimize_and_plot()
, die als erstes Argument die Lernrate entgegennimmt. Diese Funktion führt 10 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 learning rate value
lr0 = ____
optimize_and_plot(lr=lr0)