Utiliser l'optimiseur de PyTorch
Plus tôt, vous avez mis à jour manuellement le poids d'un réseau, ce qui vous a permis de comprendre comment l'entraînement fonctionne en coulisse. Cependant, cette méthode n'est pas viable pour des réseaux profonds comportant de nombreuses couches.
Heureusement, PyTorch offre l'optimiseur SGD, qui automatise ce processus efficacement en seulement quelques lignes de code. Vous allez maintenant compléter la boucle d'entraînement en mettant à jour les poids à l'aide d'un optimiseur PyTorch.
Un réseau de neurones a été créé et est fourni dans la variable model. Ce modèle a servi à exécuter une passe avant et à créer le tenseur de prédictions pred. Le tenseur one-hot est nommé target et la fonction de perte d'entropie croisée est stockée dans criterion.
torch.optim sous le nom optim, et torch.nn sous le nom nn ont déjà été chargés pour vous.
Cette activité fait partie du cours
Introduction au Deep Learning avec PyTorch
Exercice interactif pratique
Essayez cet exercice en complétant ce code d’exemple.
# Create the optimizer
optimizer = ____