Åtkomst till modellparametrar
En PyTorch-modell skapad med nn.Sequential() är en modul som innehåller nätverkets olika lager. Kom ihåg att varje lagers parametrar kan nås genom att indexera modellen direkt. I den här övningen får du öva på att komma åt parametrarna för olika linjära lager i ett neuralt nätverk.
Den här övningen är en del av kursen
Introduktion till djupinlärning med PyTorch
Övningsinstruktioner
- Hämta parametern
weightfrån det första linjära lagret. - Hämta parametern
biasfrån det andra linjära lagret.
Interaktiv övning med praktiskt arbete
Testa den här övningen genom att slutföra den här exempelkoden.
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)