CommencerCommencer gratuitement

Expérimenter avec le taux d’apprentissage

Dans cet exercice, votre objectif est de trouver le taux d’apprentissage optimal pour que l’optimiseur puisse trouver le minimum de la fonction non convexe \(x^{4} + x^{3} - 5x^{2}\) en dix étapes.

Vous expérimenterez trois valeurs différentes de taux d’apprentissage. Pour ce problème, essayez des valeurs de taux d’apprentissage comprises entre 0,001 et 0,1.

Vous disposez de la fonction optimize_and_plot() qui prend le taux d’apprentissage comme premier argument. Cette fonction exécute 10 é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 learning rate value
lr0 = ____
optimize_and_plot(lr=lr0)
Modifier et exécuter le code