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
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 = ____