一、人民币二分类实验
(验证集是用来挑选没有过拟合的模型)
Sampler:生成索引(样本的序号)
Dataset:根据索引去读取图片以及它的标签
每次装载一个batch size的数据。num_works一般可以设置为多个进程,节省训练时间。
Dataset定义数据从哪里读取以及如何读取。
getitem接受一个索引,返回样本与标签
具体的过程需要自行看代码、调试,进行理解。
(验证集是用来挑选没有过拟合的模型)
Sampler:生成索引(样本的序号)
Dataset:根据索引去读取图片以及它的标签
每次装载一个batch size的数据。num_works一般可以设置为多个进程,节省训练时间。
Dataset定义数据从哪里读取以及如何读取。
getitem接受一个索引,返回样本与标签
具体的过程需要自行看代码、调试,进行理解。
本文标题:pytorch 数据读取机制Dataloader与Dataset
本文链接:https://www.haomeiwen.com/subject/hykufhtx.html
网友评论