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

เจาะลึก method `__init__`

ใน PyTorch Lightning method __init__ คือหัวใจของการตั้งค่า LightningModule เป็นที่ที่กำหนด layer ของโมเดล พารามิเตอร์ และค่าคอนฟิกเริ่มต้นก่อนเริ่มการฝึก การแยกขั้นตอนนี้ออกมาอย่างชัดเจนช่วยให้ PyTorch Lightning ดูแลรักษาและขยายขนาดโปรเจกต์ได้ง่ายขึ้น ในแบบฝึกหัดนี้ จะเน้นที่การ initialize คลาสสำหรับโมเดล classification

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

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

ดูคอร์ส

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

  • สร้างคลาสชื่อ ClassifierModel โดย inherit จาก pl.LightningModule
  • Initialize คลาสแม่เพื่อนำความสามารถของ LightningModule มาใช้งาน

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

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

import lightning.pytorch as pl
import torch.nn as nn

# Create the class
class ClassifierModel(____):
    # Create init method
    def __init__(self, input_dim, output_dim):
        ____
        self.classifier = nn.Linear(input_dim, output_dim)
แก้ไขและรันโค้ด