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

Connected

Bài tập

Đặt mô hình ở chế độ đánh giá

Bạn đã sẵn sàng đặt mô hình ngôn ngữ vào chế độ đánh giá. Nếu mô hình không ở chế độ đánh giá khi suy luận, các tầng như batch normalization và dropout sẽ làm thay đổi hành vi của mô hình, dẫn đến chất lượng dịch không nhất quán. Hãy xây dựng vòng lặp để đánh giá mô hình!

Một số dữ liệu đã được nạp sẵn: model, eval_dataloader, accelerator, và metric.

Hướng dẫn

100 XP
  • Đặt mô hình ở chế độ đánh giá trước khi lặp qua các batch trong tập dữ liệu.
  • Gom predictions và labels trên nhiều thiết bị để tính chỉ số đánh giá bằng phương thức .gather_for_metrics() của Accelerator.
  • Tính toán chỉ số đánh giá ở cuối cùng.