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
<cours>Introduction au deep learning avec PyTorch</cours>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)