AI/3주차

2. Dataset & Dataloader

jnnwnn 2021. 8. 19. 11:35

  • Dataset: getitem에서 map-style을 통해 데이터를 어떻게 반환할지 정해줌
    • 데이터 형태에 따라 함수를 다르게 정의
    • 데이터 생성 시점에 모든 처리를 할 필요는 없다
  • transforms: 이미지 전처리, augmentation
  • Dataloader: 이를 이용하여 모델에 데이터가 들어감
 

[Pytorch] DataLoader parameter별 용도

pytorch reference 문서를 다 외우면 얼마나 편할까!!

subinium.github.io