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
Petunjuk latihan
- Akses parameter
weightdari lapisan linear pertama. - Akses parameter
biasdari 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)