zml72062的博客

转载 PyTorch 中的 torch.utils.data 解析

PyTorch数据集处理包的核心是DataLoader类。该类的构造函数签名为它构造一个可迭代对象loader，代表经过“加工”后的数据集。设置数据的加载顺序（通过修改shuffle或sampler参数）对数据进行batching处理（通过修改batch_size，，collate_fn及drop_last参数）实现multi-processloading，memorypinning等（此处不涉及）一旦构造了DataLoader对象loader，就可以用来加载数据。data和。......