CommencerCommencer 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

Introduction au deep learning avec PyTorch

Afficher le cours

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