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
Versuche dich an dieser Übung, indem du diesen Beispielcode vervollständigst.
# Try a first learning rate value
lr0 = ____
optimize_and_plot(lr=lr0)