CommencezCommencez gratuitement

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

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