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

Thiết lập Trainer

Mô hình của bạn sẽ thay thế các cụm từ phức tạp bằng diễn đạt đơn giản, dễ tiếp cận hơn để cải thiện khả năng đọc của bản dịch. Ví dụ, nó có thể đơn giản hóa cụm từ gốc "The inclement weather conditions precipitated the postponement of the outdoor event" thành "The bad weather caused the outdoor event to be delayed." Hãy xây dựng Trainer để chuẩn bị huấn luyện dịch vụ dịch ngôn ngữ của bạn! Bài tập sẽ mất một khoảng thời gian để chạy khi gọi trainer.train().

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

  • model là một mô hình Transformer
  • dataset chứa bộ dữ liệu MRPC về các câu diễn đạt lại (paraphrase)
  • Hàm compute_metrics trả về accuracy và F1 score
  • Bạn đã định nghĩa training_args ở bài trước

Hướng dẫn

100 XP
  • Truyền model vào lớp Trainer().
  • Truyền các tham số huấn luyện vào lớp Trainer().
  • Truyền một hàm tính toán metrics vào lớp Trainer().
  • In ra thiết bị mà trainer chọn.