카테고리 없음
pytorch data loader pin_memory = True
drsloth
2021. 11. 16. 08:50
https://subinium.github.io/pytorch-dataloader/
[Pytorch] DataLoader parameter별 용도
pytorch reference 문서를 다 외우면 얼마나 편할까!!
subinium.github.io
https://discuss.pytorch.org/t/when-to-set-pin-memory-to-true/19723
When to set pin_memory to true?
From the imagenet example: train_loader = torch.utils.data.DataLoader( train_dataset, batch_size=args.batch_size, shuffle=(train_sampler is None), num_workers=args.workers, pin_memory=True, sampler=train_sampler) val_loader = torch.utils.data.DataLoader( d
discuss.pytorch.org