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

Optimizers

ถึงเวลาสำรวจ optimizer แต่ละแบบที่ใช้ฝึกโมเดลได้แล้ว

มีฟังก์ชัน train_model(optimizer, net, num_epochs) เตรียมไว้ให้แล้ว ฟังก์ชันนี้รับ optimizer, โมเดล และจำนวน epoch เป็น input จากนั้นรันลูปการฝึก และแสดงค่า training loss เมื่อเสร็จสิ้น

มาใช้ train_model() เพื่อฝึกโมเดลสั้น ๆ ด้วย optimizer ต่างแบบ แล้วเปรียบเทียบผลลัพธ์กัน!

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

Deep Learning ระดับกลางด้วย PyTorch

ดูคอร์ส

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

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

import torch.optim as optim

net = Net()

# Define the SGD optimizer
optimizer = optim.____(net.parameters(), lr=0.001)

train_model(
    optimizer=optimizer,
    net=net,
    num_epochs=10,
)
แก้ไขและรันโค้ด