Začněte nyníZačněte zdarma

Přístup k parametrům modelu

Model PyTorch vytvořený pomocí nn.Sequential() je modul, který obsahuje jednotlivé vrstvy sítě. Připomeň si, že k parametrům každé vrstvy se dostaneš přímým indexováním vytvořeného modelu. V tomto cvičení si procvičíš přístup k parametrům různých lineárních vrstev neuronové sítě.

Toto cvičení je součástí kurzu

Introduction to Deep Learning with PyTorch

Zobrazit kurz

Pokyny k cvičení

  • Přistup k parametru weight první lineární vrstvy.
  • Přistup k parametru bias druhé lineární vrstvy.

Interaktivní cvičení na vyzkoušení si v praxi

Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.

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)
Upravit a spustit kód