美文网首页
pytorch 数据读取机制Dataloader与Dataset

pytorch 数据读取机制Dataloader与Dataset

作者: 米线织毛衣 | 来源:发表于2020-02-15 22:03 被阅读0次

一、人民币二分类实验

(验证集是用来挑选没有过拟合的模型)

Sampler:生成索引(样本的序号)
Dataset:根据索引去读取图片以及它的标签

每次装载一个batch size的数据。num_works一般可以设置为多个进程,节省训练时间。

Dataset定义数据从哪里读取以及如何读取。

getitem接受一个索引,返回样本与标签

具体的过程需要自行看代码、调试,进行理解。

相关文章

网友评论

      本文标题:pytorch 数据读取机制Dataloader与Dataset

      本文链接:https://www.haomeiwen.com/subject/hykufhtx.html