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

연습 문제

훈련 DataLoader 만들기

데이터셋을 분할했으니, 이제 학습 중에 배치 단위로 데이터를 제공할 데이터 로더를 정의해야 해요. DataLoader는 데이터를 메모리에 효율적으로 적재하고, 더 나은 일반화를 위해 셔플을 허용해요. 이번 연습에서는 train_dataloader 메서드를 완성해 보세요.

지침

100 XP
  • DataLoader를 임포트하세요.
  • 더 나은 일반화를 위해 셔플을 활성화하고, self.train_data를 로드하는 DataLoader를 반환하세요.