Bắt đầu ngayBắt đầu miễn phí

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

Bài tập này là một phần của khóa học

Huấn luyện Mô hình AI Hiệu quả với PyTorch

Xem khóa học

Hướng dẫn bài tập

  • 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.

Bài tập tương tác thực hành trực tiếp

Hãy thử làm bài tập này bằng cách hoàn thành đoạn mã mẫu này.

trainer = Trainer(
    # Pass in the model
    model=____,
    # Input the training arguments
    args=____,
    train_dataset=dataset["train"],
    eval_dataset=dataset["validation"],
    # Pass in a function to compute metrics
    compute_metrics=____,
)

trainer.train()

# Print the device that the trainer chooses
print(trainer.args.____)
Chỉnh sửa và Chạy Mã