首页 > 其他 > 详细

Dataset、DataLoader、DataLoaderIter

时间:2020-09-12 22:50:57      阅读:188      评论:0      收藏:0      [点我收藏+]

一、Dataset

对单个样本的处理

1、__getitem__:

返回一条数据或一个样本,obj[index] == obj.__getitem__()

2、__len__:

返回整个数据集中样本的数量,len(obj) == obj.__len__()

二、DataLoader

对batch个样本的处理

主要完成两件事:

a、定义了一堆成员变量,后面赋值给DataLoaderIter

b、将自身传入DataLoaderIter

三、DataLoaderIter

Dataset装入DataLoader,DataLoader装入DataLoaderIter

 

for data in DataLoader:

a、调用DataLoader的__iter__()得到迭代器DataLoaderIter

b、循环调用DataLoaderIter的__next__函数

 

Dataset、DataLoader、DataLoaderIter

原文:https://www.cnblogs.com/liujianing/p/13658261.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!