1. Học hỏi
    2. /
  2. Khoa Học
    3. /
  3. Deep Learning nâng cao với PyTorch
Connected

Bài tập

Mạng GRU

Bên cạnh LSTM, một biến thể mạng nơ-ron hồi tiếp phổ biến khác là Gated Recurrent Unit (GRU). Điểm hấp dẫn của nó nằm ở sự đơn giản: các cell GRU cần ít tính toán hơn cell LSTM nhưng thường cho hiệu năng tương đương.

Đoạn mã bạn nhận được là phần định nghĩa mô hình RNN mà bạn đã viết trước đó. Nhiệm vụ của bạn là chỉnh sửa để mô hình này trở thành mạng GRU. torch và torch.nn as nn đã được nhập sẵn cho bạn.

Hướng dẫn

100 XP
  • Cập nhật định nghĩa mô hình RNN để thu được mạng GRU; gán tầng GRU cho self.gru.