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

กำหนดค่า optimizer

เมื่อมี training logic แล้ว ขั้นตอนต่อไปคือการระบุวิธี optimize parameter ของโมเดล

ในแบบฝึกหัดนี้ จะต้องเติม method configure_optimizers ใน PyTorch Lightning module ที่ใช้สำหรับงานจำแนกภาพ เป้าหมายคือตั้งค่า optimizer สำหรับอัปเดต parameter ของโมเดลระหว่างการฝึก โดยใช้ Adam optimizer ที่มี learning rate เท่ากับ 1e-3

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

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

ดูคอร์ส

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

  • สร้าง Adam optimizer โดยใช้ parameter ของโมเดล และกำหนด learning rate เป็น 1e-3

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

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

import torch

def configure_optimizers(self):
  	# Create an Adam optimizer for model parameters
    optimizer = ____ 
    return optimizer
แก้ไขและรันโค้ด