PyTorch笔记 Dataset和Dataloader

在PyTorch中,数据集会用一个类来表示,在训练时用Dataloader产生一个batch的数据。
Dataset是pytorch中用来表示数据集的一个抽象类。
Dataloader作为一个迭代器。
Dataset:至少覆盖写下面两个方法,返回数据前可以进行适当的处理
1__len__:数据集大小
2__getitem__:实现这个方法后,可以通过下标的方式(dataset[i])获得i个数据
DataLoader:迭代器,最基本的使用是传入一个Dataset对象,它就会根据参数batch_size生成一个batch的数据

你可能感兴趣的:(笔记,python)