pytorch1 [PyTorch] Dataset과 Dataloader는 왜 사용할까 결론을 먼저 말하자면 "Customizing" 하려고이다. 딥러닝 모델을 학습시킬때 데이터셋을 가져와야 하는데 한번에 모든 데이터를 가져오면 메모리가 터져버리는 일이 생긴다.그래서 batch를 나눠서 모델이 입력을 해주는데 어떤 데이터를 가져오고, 어떤 형태로 가져오고, 한 번에 얼마나 가져올지 customzing 하기 위해 torch에 dataset과 dataloader를 사용한다. DatasetDataset class는 전체 dataset을 구성하는 단계이다. input으로 전체 x와 y를 텐서로 넣어준다.__init__(self): 여기서 필요한 변수들을 선언한다.__len__(self) : x와 y의 데이터의 길이는 같으니까 편의상 len(y)를 통해 데이터의 크기를 리턴한다.__getitem__.. 2021. 3. 23. 이전 1 다음
