qq_36044523的博客

DataLoader(object)类：DataLoader 下有__init__，__setattr__，__iter__，__len__等四个函数，详情如下。__init__ 函数1.dataset(Dataset)，数据读取接口（比如torchvision.datasets.ImageFolder）或者自定义的数据接口的输出，该输出是torch.utils.data.Dataset类的对象(或者继承自该类的自定义类的对象)。2.batch_size (int, optional)，批训练数据