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
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)