Pytorch Dataset和DataLoader 加载训练数据
Dataset 基类torch.utils.data.Dataset 为数据集的基类, 继承这个基类,我们能够非常快速的实现对数据的加载。我们要实现自己加载数据的类,并继承于Dataset 这个类,重载类的成员函数1、__1en__方法, 能够实现通过全局的len()方法获取其中的元素个数;2、getitem 方法,能够通过传入索引的方式获取数据,例如通过dataset[i]获取其中的第 i条数据。from torch.utils.data import Dataset, DataLoader#
原创
2022-05-19 15:37:33 ·
477 阅读 ·
0 评论