1. Học hỏi
    2. /
  2. Khoa Học
    3. /
  3. Nhập môn Deep Learning với PyTorch
Connected

Bài tập

Đếm số lượng tham số

Các mô hình deep learning nổi tiếng vì có rất nhiều tham số. Càng nhiều tham số thì độ phức tạp tính toán càng cao và thời gian huấn luyện càng dài, nên người làm deep learning cần biết mô hình của mình có bao nhiêu tham số.

Trong bài tập này, bạn sẽ tự tính số lượng tham số bằng tay trước. Sau đó, bạn sẽ kiểm tra lại kết quả của mình bằng phương thức .numel().

Hướng dẫn 1/2

undefined XP
    1
    2

Câu hỏi

Tính thủ công số lượng tham số của mô hình dưới đây. Mô hình có bao nhiêu tham số? Dùng console như một máy tính.

model = nn.Sequential(nn.Linear(9, 4),
                      nn.Linear(4, 2),
                      nn.Linear(2, 1))

Các phương án trả lời