1. 학습
  2. /
  3. 강의
  4. /
  5. PyTorch로 배우는 딥러닝 입문

Connected

연습 문제

DataLoader 사용하기

DataLoader 클래스는 대용량 데이터셋을 효율적으로 처리하는 데 필수적이에요. 학습을 빠르게 하고, 메모리 사용을 최적화하며, 그래디언트 업데이트를 안정화해 deep learning 모델의 성능을 높여 줍니다.

이제 이전 연습 문제에서 만든 dataset으로 PyTorch DataLoader를 생성하고, 어떻게 동작하는지 확인해 보세요.

지침

100 XP
  • 필요한 모듈을 임포트하세요.
  • dataset을 사용해 DataLoader를 만들고, 배치 크기는 2로 설정하며 셔플을 활성화하세요.
  • DataLoader를 순회(iterate)하면서 각 배치의 입력과 레이블을 출력하세요.