CommencerCommencer gratuitement

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

Afficher le cours

Exercice interactif pratique

Essayez cet exercice en complétant cet exemple de code.

# Create the optimizer
optimizer = ____
Modifier et exécuter le code