美文网首页
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