目录
1.torch.utils.data.DataLoader概念介绍
2.torch.utils.data.DataLoader参数介绍
DataLoader:[batch_sizeN, C, H, W]+[Target1, Target2, ......., TargetN] ?
1.torch.utils.data.DataLoader概念介绍
Data loader. Combines a dataset and a sampler, and provides an iterable over the given dataset.
- DataLoader是Pytorch中用来处理模型输入数据的一个工具类。
- 组合了数据集(dataset) + 采样器(sampler),
- 并在数据集上提供单线程或多线程(num_workers )的可迭代对象。
只要用PyTorch训练模型,往往都离不开torch.utils.data.DataLoader接口,其主要作用就是将自定义的数据读取接口的输出或者PyTorch已有的数据接口的输入