AI beginner by 비전공자
[Pytorch] Dataset , Dataloader 뜻과 사용법
Semi_AI_Expert
2022. 10. 12. 16:41
Dataset , Dataloader :학습 모델에 맞게 학습데이터 모음과 테스트데이터 모음을 변경해주는 역할
1. torch.utils.data.Dataset
사용가능 데이터 : pytorch library에 있는 data set & 미리 준비된 data set 도 사용가능
csv파일의 경우 : Dataset class 를 상속받아 구현 가능