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

연습 문제

LightningDataModule로 데이터 나누기

LightningDataModule의 setup 메서드를 완성해 보세요. 올바르게 데이터셋을 분할하면 모델은 한 부분으로 학습하고 다른 부분으로 검증하여 과적합을 방지할 수 있습니다.

dataset은 이미 미리 임포트되어 있습니다.

지침

100 XP
  • 학습과 검증을 위해 데이터셋을 분할할 수 있도록 random_split을 임포트하세요.
  • random_split을 사용해 데이터셋을 학습용(80%)과 검증용(20%)으로 나누세요.