Pytorch的Dataset与Dataloader之间的关系

Posted 2023-03-04

参考技术A 在 PyTorch 中，我们的数据集往往会用一个类去表示，在训练时用 Dataloader 产生一个 batch 的数据

一般我们需要自己去实现一个datasets对象，传入到dataloader中；然后dataloader内部使用yeild返回每一次batch大小的数据；
pytorch 的数据加载到模型的操作顺序是这样的：

① 创建一个 Dataset 对象
② 创建一个 DataLoader 对象
③ 循环这个 DataLoader 对象，将img, label加载到模型中进行训练

具体参内容考见下面的四个链接即可。