CommencerCommencez gratuitement

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>
Voir le 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)
Modifier et exécuter le code