ÎncepețiÎncepe gratuit

Accesarea parametrilor modelului

Un model PyTorch creat cu nn.Sequential() este un modul care conține diferitele straturi ale rețelei tale. Reține că parametrii fiecărui strat pot fi accesați prin indexarea directă a modelului creat. În acest exercițiu, vei exersa accesarea parametrilor diferitelor straturi liniare ale unei rețele neuronale.

Acest exercițiu face parte din cursul

Introducere în Deep Learning cu PyTorch

Vezi cursul

Instrucțiuni pentru exercițiu

  • Accesează parametrul weight al primului strat liniar.
  • Accesează parametrul bias al celui de-al doilea strat liniar.

Exercițiu interactiv practic

Încearcă acest exercițiu completând acest cod de exemplu.

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)
Editează și rulează codul