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

Sử dụng DataLoader

Lớp DataLoader rất quan trọng để xử lý hiệu quả các tập dữ liệu lớn. Nó tăng tốc quá trình huấn luyện, tối ưu hóa việc sử dụng bộ nhớ và giúp cập nhật gradient ổn định hơn, từ đó làm mô hình deep learning hiệu quả hơn.

Bây giờ, bạn sẽ tạo một DataLoader của PyTorch từ dataset ở bài trước và xem nó hoạt động như thế nào.

Hướng dẫn

100 XP
  • Import mô-đun cần thiết.
  • Tạo một DataLoader từ dataset, đặt kích thước batch là hai và bật xáo trộn (shuffling).
  • Lặp qua DataLoader và in ra từng batch của input và label.