MulaiMulai sekarang secara gratis

Mengakses parameter model

Model PyTorch yang dibuat dengan nn.Sequential() adalah modul yang memuat berbagai lapisan jaringan Anda. Ingat bahwa setiap parameter lapisan dapat diakses dengan melakukan pengindeksan langsung pada model yang dibuat. Dalam latihan ini, Anda akan berlatih mengakses parameter dari berbagai lapisan linear pada sebuah neural network.

Latihan ini adalah bagian dari kursus

Pengantar Deep Learning dengan PyTorch

Lihat Kursus

Petunjuk latihan

  • Akses parameter weight dari lapisan linear pertama.
  • Akses parameter bias dari lapisan linear kedua.

Latihan interaktif praktis

Cobalah latihan ini dengan menyelesaikan kode contoh berikut.

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)
Edit dan Jalankan Kode