CommencerCommencer gratuitement

Mise à jour manuelle des poids

Maintenant que vous savez comment accéder aux poids et aux biais, vous allez effectuer manuellement le travail de l’optimiseur PyTorch. Bien que cette opération soit automatisée par PyTorch, la réaliser manuellement renforce l’intuition sur les mécanismes d’apprentissage et d’ajustement des modèles. Cette compréhension sera précieuse lors du débogage ou du réglage des réseaux neuronaux.

Un réseau neuronal à trois couches a été créé et stocké en tant que variable model. Ce réseau a été utilisé pour une passe avant et la perte et ses dérivées ont été calculées. Un taux d’apprentissage par défaut, lr, a été choisi pour échelonner les gradients lors de la mise à jour.

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.

weight0 = model[0].weight
weight1 = model[1].weight
weight2 = model[2].weight

# Access the gradients of the weight of each linear layer
grads0 = ____
grads1 = ____
grads2 = ____
Modifier et exécuter le code