Dataset、DataLoader、DataLoaderIter

技术文章 1年前 (2020) 完美者
466 0

标签:变量   item   数据   set   batch   get   赋值   bat   ade   

一、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

标签:变量   item   数据   set   batch   get   赋值   bat   ade   

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

版权声明:完美者 发表于 2020-09-17 23:20:15。
转载请注明:Dataset、DataLoader、DataLoaderIter | 完美导航

暂无评论

暂无评论...