ComenzarEmpieza gratis

Acceso a los parámetros del modelo

Un modelo PyTorch creado con nn.Sequential() es un módulo que contiene las diferentes capas de tu red. Recuerda que se puede acceder a cada parámetro de la capa indexando directamente el modelo creado. En este ejercicio, practicarás el acceso a los parámetros de diferentes capas lineales de una red neuronal.

Este ejercicio forma parte del curso

Introducción al aprendizaje profundo con PyTorch

Ver curso

Instrucciones del ejercicio

  • Accede al parámetro weight de la primera capa lineal.
  • Accede al parámetro bias de la segunda capa lineal.

Ejercicio interactivo práctico

Prueba este ejercicio completando el código de muestra.

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)
Editar y ejecutar código