Pytorch笔记:DataLoader,Dataset和Sampler
简介:在 PyTorch 中,我们的数据集往往会用一个类去表示,在训练时用 Dataloader 产生一个 batch 的数据。简单说,用一个类 抽象地表示数据集,而 Dataloader 作为迭代器,每次产生一个 batch 大小的数据,节省内存。pytorch中加载数据的顺序是:①创建一个dataset对象②创建一个dataloader对象③循环dataloader对象,将data,label拿到模型中去训练Datasettorch.utils.data.Dataset是一个抽象类,是Py
原创
2021-04-20 22:17:51 ·
4024 阅读 ·
0 评论