1. Học hỏi
    2. /
  2. Khoa Học
    3. /
  3. Huấn luyện Mô hình AI Hiệu quả với PyTorch
Connected

Bài tập

Chuẩn bị tập dữ liệu cho huấn luyện phân tán

Bạn đã tiền xử lý một tập dữ liệu cho hệ thống nông nghiệp chính xác nhằm giúp nông dân theo dõi sức khỏe cây trồng. Giờ bạn sẽ tải dữ liệu bằng cách tạo một DataLoader và đặt dữ liệu lên GPU để huấn luyện phân tán, nếu có GPU sẵn có. Lưu ý: bài tập thực tế chạy trên CPU, nhưng mã dùng cho CPU và GPU là như nhau.

Một số dữ liệu đã được nạp sẵn:

  • Một dataset mẫu với ảnh nông nghiệp
  • Lớp Accelerator từ thư viện accelerate
  • Lớp DataLoader

Hướng dẫn

100 XP
  • Tạo dataloader cho dataset đã định nghĩa sẵn.
  • Đặt dataloader lên các thiết bị sẵn có bằng đối tượng accelerator.