ПочатиПочніть безкоштовно

Доступ до параметрів моделі

Модель PyTorch, створена за допомогою nn.Sequential(), — це модуль, що містить різні шари вашої мережі. Пам'ятайте, що параметри кожного шару можна отримати, індексуючи створену модель безпосередньо. У цій вправі ви попрактикуєтеся в доступі до параметрів різних лінійних шарів нейронної мережі.

Ця вправа є частиною курсу

Вступ до Deep Learning з 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)
Редагувати та запускати код