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

การ implement validation step

เมื่อเทรนโมเดลโครงข่ายประสาทเทียมแล้ว เราต้องติดตามประสิทธิภาพของโมเดลระหว่างการเทรน ให้ใช้ PyTorch Lightning implement เมธอด validation_step() เพื่อคำนวณและบันทึกค่า validation loss ในแต่ละ epoch

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

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

ดูคอร์ส

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

  • คำนวณการพยากรณ์โดยใช้โมเดลกับ input batch
  • คำนวณค่า validation loss ด้วย F.cross_entropy()
  • บันทึกค่า validation loss ด้วย self.log() ในชื่อ val_loss

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

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

import torch.nn.functional as F

def validation_step(self, batch, batch_idx):
    x, y = batch
    # Compute predictions using the model
    preds = ____(x)
    # Calculate validation loss
    loss = F.____(preds, y)
    # Log the validation loss
    self.____('val_loss', loss)
แก้ไขและรันโค้ด