CommencezCommencez gratuitement

Mettre à jour les poids manuellement

Maintenant que vous savez comment accéder aux poids et aux biais, vous allez effectuer manuellement le travail de l'optimiseur PyTorch. Même si PyTorch automatise cette étape, la pratiquer à la main vous aide à développer une intuition sur la façon dont les modèles apprennent et s'ajustent. Cette compréhension sera utile pour le débogage ou l'ajustement fin de réseaux de neurones.

Un réseau de neurones à trois couches a été créé et stocké dans la variable model. Ce réseau a été utilisé pour une propagation avant et la perte ainsi que ses dérivées ont été calculées. Un taux d'apprentissage par défaut, lr, a été choisi pour mettre à l'échelle les gradients lors de la mise à jour.

Cette activité fait partie du cours

Introduction au Deep Learning avec PyTorch

Voir le cours

Exercice interactif pratique

Essayez cet exercice en complétant ce code d’exemple.

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