เริ่มต้นใช้งานเริ่มต้นใช้งานได้ฟรี

Mixed precision training ด้วย Trainer

ต้องการทำให้อินเทอร์เฟซการเทรนสำหรับ mixed precision training เรียบง่ายขึ้น แทนที่จะใช้ training loop กับ Accelerator จึงเลือกสร้าง Trainer เนื่องจากโมเดลไม่จำเป็นต้องใช้ training loop แบบกำหนดเอง ตั้งค่า Trainer ให้ใช้ mixed precision training ได้เลย โปรดทราบว่าการรันโค้ดอาจใช้เวลาสักครู่เมื่อเรียก trainer.train()

แบบฝึกหัดนี้เป็นส่วนหนึ่งของหลักสูตร

การเทรน AI Model อย่างมีประสิทธิภาพด้วย PyTorch

ดูคอร์ส

คำแนะนำการฝึกหัด

  • เปิดใช้งาน mixed precision training สำหรับ Trainer

แบบฝึกหัดเชิงโต้ตอบแบบลงมือทำ

ลองทำแบบฝึกหัดนี้โดยเติมโค้ดตัวอย่างนี้ให้สมบูรณ์

training_args = TrainingArguments(
    output_dir="./results",
    evaluation_strategy="epoch",
    # Enable mixed precision training
    ____=____
)
trainer = Trainer(model=model,
                  args=training_args,
                  train_dataset=dataset["train"],
                  eval_dataset=dataset["validation"],
                  compute_metrics=compute_metrics)
trainer.train()
แก้ไขและรันโค้ด