AI/3주차
2. Dataset & Dataloader
jnnwnn
2021. 8. 19. 11:35
- Dataset: getitem에서 map-style을 통해 데이터를 어떻게 반환할지 정해줌
- 데이터 형태에 따라 함수를 다르게 정의
- 데이터 생성 시점에 모든 처리를 할 필요는 없다
- transforms: 이미지 전처리, augmentation
- Dataloader: 이를 이용하여 모델에 데이터가 들어감
- Data의 batch를 만들어줌
- https://subinium.github.io/pytorch-dataloader/
[Pytorch] DataLoader parameter별 용도
pytorch reference 문서를 다 외우면 얼마나 편할까!!
subinium.github.io