分类：torch.utils.data.DataLoader

2020
09-25

python torch.utils.data.DataLoader使用方法

PyTorch中数据读取的一个重要接口是torch.utils.data.DataLoader，该接口定义在dataloader.py脚本中，只要是用PyTorch来训练模型基本都会用到该接口，该接口主要用来将自定义的数据读取接口的输出或者PyTorch已有的数据读取接口的输入按照batchsize封装成Tensor，后续只需要再包装成Variable即可作为模型的输入，因此该接口有点承上启下的作用，比较重要。数据加载器，结合了数据集和取样器，并且可以提供多个线程处理数据集。在... 继续阅读 >