pytorch 数据读取之DataLoader与Dataset

深度之眼学习心得 实验:人民币二分类,训练一个二分类器,可以区别出第四套人民币的1元与100元。 模型训练的5大步骤,数据,模型,损失函数,优化器,迭代训练 本文只探讨数据。 数据:数据收集---img,label            数据分割---train,valid,test            数据读取---DataLoader(Sampler(生成索引),Dataset(根据索引拿到
相关文章
相关标签/搜索