Dostęp do parametrów modelu
Model PyTorch utworzony za pomocą nn.Sequential() to moduł zawierający poszczególne warstwy sieci. Pamiętaj, że dostęp do parametrów każdej warstwy uzyskuje się przez bezpośrednie indeksowanie modelu. W tym ćwiczeniu przećwiczysz dostęp do parametrów różnych warstw liniowych sieci neuronowej.
To ćwiczenie jest częścią kursu
Wprowadzenie do uczenia głębokiego z PyTorch
Instrukcje do ćwiczenia
- Uzyskaj dostęp do parametru
weightpierwszej warstwy liniowej. - Uzyskaj dostęp do parametru
biasdrugiej warstwy liniowej.
Interaktywne ćwiczenie praktyczne
Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.
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)