บันทึกและโหลดโมเดล

บริษัทผู้ผลิตแห่งหนึ่งต้องการจำแนกประเภทสินค้าจากภาพ เพื่อเลือกบรรจุภัณฑ์สำหรับการจัดส่งที่เหมาะสม หลังจากฝึกโมเดลที่มีความแม่นยำสูงใน PyTorch เรียบร้อยแล้ว ขั้นตอนต่อไปคือบันทึกโมเดลพร้อมค่าน้ำหนักที่ฝึกไว้ เพื่อนำไปใช้ในอนาคตและแชร์ให้ทีมงานสามารถโหลดไปใช้ได้อย่างราบรื่น

มีการ import torch และ torch.nn เป็น nn แล้ว โมเดลที่ฝึกไว้พร้อมใช้งานในชื่อ model และสถาปัตยกรรมของโมเดลอยู่ในคลาส ManufacturingCNN

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

Deep Learning สำหรับภาพด้วย PyTorch

ดูคอร์ส

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

บันทึกโมเดลที่ฝึกแล้วเป็นไฟล์ ModelCNN.pth โดยอย่าลืมบันทึกค่าน้ำหนัก ไม่ใช่แค่สถาปัตยกรรมของโมเดล
สร้าง instance ของโมเดลชื่อ loaded_model จากคลาส ManufacturingCNN()
โหลดค่าน้ำหนักจากไฟล์ ModelCNN.pth เข้าสู่ loaded_model โดยส่งค่าน้ำหนักไปยัง .load_state_dict()

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

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

# Save the model
torch.____(model.____, ____)

# Create a new model
loaded_model = ____

# Load the saved model
loaded_model.____(torch.____('ModelCNN.pth'))
print(loaded_model)

แก้ไขและรันโค้ด