CommencerCommencer gratuitement

Accès aux paramètres du modèle

Un modèle PyTorch créé avec nn.Sequential() est un module qui contient les différentes couches de votre réseau. Rappelez-vous que chaque paramètre de couche est accessible en indexant directement le modèle créé. Dans cet exercice, vous vous entraînerez à accéder aux paramètres de différentes couches linéaires d’un réseau neuronal.

Cet exercice fait partie du cours

Introduction au deep learning avec PyTorch

Afficher le cours

Instructions

  • Accédez au paramètre weight de la première couche linéaire.
  • Accédez au paramètre bias de la deuxième couche linéaire.

Exercice interactif pratique

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

model = nn.Sequential(nn.Linear(16, 8),
                      nn.Linear(8, 2)
                     )

# Access the weight of the first linear layer
weight_0 = ____
print("Weight of the first layer:", weight_0)

# Access the bias of the second linear layer
bias_1 = ____
print("Bias of the second layer:", bias_1)
Modifier et exécuter le code