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
<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 learning rate value
lr0 = ____
optimize_and_plot(lr=lr0)