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
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)