1. 학습
    2. /
  2. 강의
    3. /
  3. PyTorch Lightning으로 만드는 확장 가능한 AI 모델
Connected

연습 문제

Lightning으로 모델 학습 최적화하기

ModelCheckpoint와 EarlyStopping 같은 자동화 기법을 적용하면, 불필요한 계산을 줄이면서도 최적의 성능을 내는 파라미터를 선택할 수 있어요.

이 데이터셋은 Osmanya MNIST의 일부로, 확장 가능한 AI 학습 기법이 효율성과 정확도를 크게 높일 수 있는 실제 사례를 제공합니다.

OsmanyaDataModule과 ImageClassifier는 미리 정의되어 있습니다.

지침

100 XP
  • 모델 체크포인트와 조기 종료에 사용할 콜백을 임포트하세요.
  • ModelCheckpoint와 EarlyStopping 콜백을 사용해 모델을 학습하세요.