Utilisation de l’optimiseur PyTorch
Plus tôt, vous avez mis à jour manuellement le poids d’un réseau, ce qui vous a permis de comprendre le fonctionnement de l’apprentissage en coulisses. Toutefois, cette méthode n’est pas évolutive pour les réseaux profonds comportant de nombreuses couches.
Heureusement, PyTorch propose l’optimiseur SGD, qui automatise efficacement ce processus en quelques lignes de code. Vous allez maintenant terminer la boucle d’apprentissage en mettant à jour les poids à l’aide d’un optimiseur PyTorch.
Un réseau neuronal a été créé et fourni en tant que variable model
. Ce modèle a été utilisé pour effectuer une passe avant et créer le tenseur de prédictions pred
. Le tenseur encodé en one-hot est nommé target
et la fonction de perte d’entropie croisée est stockée en tant que criterion
.
Nous avons déjà chargé torch.optim
sous le nom optim
, et torch.nn
sous le nom nn
pour vous.
Cet exercice fait partie du cours
Introduction au deep learning avec PyTorch
Exercice interactif pratique
Essayez cet exercice en complétant cet exemple de code.
# Create the optimizer
optimizer = ____