1. 学ぶ
    2. /
  2. コース
    3. /
  3. PyTorch Lightning으로 만드는 확장 가능한 AI 모델
Connected

演習

init 메서드 마스터하기

PyTorch Lightning에서 __init__ 메서드는 LightningModule 설정의 핵심입니다. 이곳에서 학습 전에 모델 레이어, 하이퍼파라미터, 초기 설정을 정의합니다. 이런 설정 단계를 깔끔하게 분리하면 프로젝트를 유지보수하고 확장하기가 훨씬 쉬워집니다. 이번 연습에서는 분류 모델 클래스를 초기화하는 데 집중해 보겠습니다.

指示

100 XP
  • pl.LightningModule을 상속하는 ClassifierModel 클래스를 만드세요.
  • LightningModule의 기능을 활용할 수 있도록 부모 클래스를 초기화하세요.