НачатьНачать бесплатно

Доступ к параметрам модели

Модель PyTorch, созданная с помощью nn.Sequential(), — это модуль, содержащий различные слои сети. Напомним, что к параметрам каждого слоя можно обратиться, напрямую индексируя созданную модель. В этом упражнении вы потренируетесь получать доступ к параметрам различных линейных слоёв нейронной сети.

Это упражнение является частью курса

Введение в глубокое обучение с PyTorch

Посмотреть курс

Инструкции к упражнению

  • Получите доступ к параметру weight первого линейного слоя.
  • Получите доступ к параметру bias второго линейного слоя.

Интерактивное практическое упражнение

Попробуйте выполнить это упражнение, дополнив этот пример кода.

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)
Редактировать и запускать код