[PyTorch] torch.utils.data, torchvision

모델을 학습하기 위해서는 데이터를 모델에 지속적으로 전달해야 한다. 데이터를 하나씩 전달하지 않고 원하는 batch size로 묶어서 전달하거나 더 효율적인 학습을 위해 데이터를 규칙에 따라 정렬하거나 섞어야 할 때 사용하는 모듈이 DataLoader이다. 형태로 쓴다. 매개변수가 굉장히 많은데 중요한 것만 보면 - data : 다룰 데이터를 넣어주면 된다. - batch_size : 데이터를 batch_size 개수만큼 묶는다는 의미이다. - shuffle : 데이터를 매 epoch에서 다시 섞을지를 결정한다. default = False이다. - num_workers : 데이터를 묶을 때 사용할 프로세스의 수 - drop_last : 데이터세트 크기가 배치 크기로 나눌 수 없는 경우에 나누고..........

원문링크 : [PyTorch] torch.utils.data, torchvision