Truy cập các tham số của mô hình
Một mô hình PyTorch được tạo với nn.Sequential() là một module chứa các lớp khác nhau của mạng. Hãy nhớ rằng bạn có thể truy cập tham số của từng lớp bằng cách đánh chỉ số trực tiếp vào mô hình đã tạo. Trong bài tập này, bạn sẽ thực hành truy cập các tham số của những lớp linear khác nhau trong một mạng nơ-ron.
Bài tập này là một phần của khóa học
Nhập môn Deep Learning với PyTorch
Hướng dẫn bài tập
- Truy cập tham số
weightcủa lớp linear đầu tiên. - Truy cập tham số
biascủa lớp linear thứ hai.
Bài tập tương tác thực hành trực tiếp
Hãy thử làm bài tập này bằng cách hoàn thành đoạn mã mẫu này.
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)