pytorch 数据读取机制Dataloader与Dataset

2020-02-15  本文已影响0人  米线织毛衣

一、人民币二分类实验

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

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

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

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

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

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

上一篇下一篇

猜你喜欢

热点阅读