init 메서드 마스터하기
PyTorch Lightning에서 __init__ 메서드는 LightningModule 설정의 핵심입니다. 이곳에서 학습 전에 모델 레이어, 하이퍼파라미터, 초기 설정을 정의합니다. 이런 설정 단계를 깔끔하게 분리하면 프로젝트를 유지보수하고 확장하기가 훨씬 쉬워집니다. 이번 연습에서는 분류 모델 클래스를 초기화하는 데 집중해 보겠습니다.
이 연습은 강의의 일부입니다
PyTorch Lightning으로 만드는 확장 가능한 AI 모델
연습 안내
pl.LightningModule을 상속하는ClassifierModel클래스를 만드세요.- 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)