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

Tự động gán thiết bị với Accelerator

Mô hình AI hội thoại của bạn cần huấn luyện trên một tập dữ liệu khổng lồ, nên bạn quyết định chuyển mô hình sang GPU. Bạn đang tận dụng Accelerator để tự động gán thiết bị. Lưu ý: bài tập này thực tế chạy trên CPU, nhưng mã không đổi khi chạy trên GPU.

Một mô hình dựa trên BERT đã được nạp sẵn dưới tên model.

Hướng dẫn

100 XP
  • Khai báo một đối tượng accelerator bằng cách khởi tạo lớp phù hợp.
  • Dùng đối tượng accelerator để chuẩn bị mô hình cho huấn luyện phân tán với GPU.