Expérimenter avec la dynamique (momentum)
Dans cet exercice, votre objectif est de trouver la dynamique (momentum) optimale qui permettra à l’optimiseur de trouver le minimum de la fonction non convexe suivante \(x^{4} + x^{3} - 5x^{2}\) en 20 étapes. Vous allez expérimenter deux valeurs de momentum différentes. Pour ce problème, le taux d’apprentissage est fixé à 0,01.
Vous disposez de la fonction optimize_and_plot()
qui accepte en entrée le paramètre momentum
. Cette fonction exécute 20 étapes de l’optimiseur SGD et affiche les résultats.
Cet exercice fait partie du cours
Introduction au deep learning avec PyTorch
Exercice interactif pratique
Essayez cet exercice en complétant cet exemple de code.
# Try a first value for momentum
mom0 = ____
optimize_and_plot(momentum=mom0)