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 merupakan bagian dari kursus
Pengantar Deep Learning dengan PyTorch
Instruksi latihan
- Akses parameter
weightdari lapisan linear pertama. - Akses parameter
biasdari lapisan linear kedua.
Latihan interaktif langsung praktik
Cobalah latihan ini dengan melengkapi kode contoh ini.
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)