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

การรัน Lightning Trainer

เมื่อกำหนดโมเดลเรียบร้อยแล้ว ขั้นตอนต่อไปคือการตั้งค่า Lightning Trainer เพื่อรันลูปการฝึก แบบฝึกหัดนี้จะช่วยให้เข้าใจวิธีเชื่อมโยงโมเดล data loader และการตั้งค่าการฝึกเข้าด้วยกัน ซึ่งเป็นพื้นฐานสำคัญก่อนก้าวไปสู่การฝึกในระดับที่ซับซ้อนยิ่งขึ้น

torch และ lightning.pytorch ซึ่ง import มาในชื่อ pl ได้ถูกโหลดไว้ให้แล้ว รวมถึง model ที่ initialize ด้วยคลาสที่กำหนดในแบบฝึกหัดก่อนหน้า

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

AI Models แบบ Scalable ด้วย PyTorch Lightning

ดูคอร์ส

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

  • ตั้งค่า Trainer ให้รัน 3 epoch
  • เริ่มการฝึกโดยใช้ model ที่กำหนดไว้แล้ว

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

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

from lightning.pytorch import Trainer
# Configure the Trainer with max epochs
trainer = ____(max_epochs=____, devices=1)
# Start training with the predefined model
trainer.____(____, train_dataloader, val_dataloader)
แก้ไขและรันโค้ด