Expérimenter avec le momentum
Dans cet exercice, votre objectif est de trouver le momentum optimal pour que l'optimiseur atteigne le minimum de la fonction non convexe suivante : \(x^{4} + x^{3} - 5x^{2}\) en 20 étapes. Vous testerez deux valeurs différentes de momentum. Pour ce problème, le taux d'apprentissage est fixé à 0,01.
On met à votre disposition la fonction optimize_and_plot() qui accepte en entrée le paramètre momentum. Cette fonction exécutera 20 étapes de l'optimiseur SGD et affichera 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 value for momentum
mom0 = ____
optimize_and_plot(momentum=mom0)