Expérimenter avec le taux d'apprentissage
Dans cet exercice, votre objectif est de trouver le taux d'apprentissage optimal pour que l'optimiseur atteigne le minimum de la fonction non convexe \(x^{4} + x^{3} - 5x^{2}\) en dix étapes.
Vous allez essayer trois valeurs différentes de taux d'apprentissage. Pour ce problème, testez des valeurs entre 0,001 et 0,1.
La fonction optimize_and_plot() vous est fournie et prend le taux d'apprentissage comme premier argument. Cette fonction exécute 10 étapes de l'optimiseur SGD et affiche les résultats.
Cette activité fait partie du cours
Introduction au Deep Learning avec PyTorch
Exercice interactif pratique
Essayez cet exercice en complétant ce code d’exemple.
# Try a first learning rate value
lr0 = ____
optimize_and_plot(lr=lr0)