PyTorch: torch.utils.data.DataLoader

danbibibi·2022년 1월 21일
0

PyTorch 🔥

목록 보기
5/20

torch.utils.data.DataLoader

Dataloader class는 batch기반의 딥러닝모델 학습을 위해서 mini batch를 만들어주는 역할을 한다. 여러 옵션(데이터 묶기, 섞기, 병렬처리)을 통해 batch를 만들어주어, 학습 시에 data를 batch만큼 편하게 꺼내어 사용할 수 있다.

DataLoader(dataset, batch_size=128, shuffle=False, sampler=None,
           batch_sampler=None, num_workers=0, collate_fn=None,
           pin_memory=False, drop_last=False, timeout=0,
           worker_init_fn=None)
profile
블로그 이전) https://danbibibi.tistory.com

0개의 댓글