自定义Map类型的Dataset只需要定义类,并继承torch.utils.data.Dataset方法即可,但要实现两个重要方法:__getitem__(self, index)和__len__(self) 个人比较推荐使用这种Dataset 例如: fromtorch.utils.dataimportDatasetfromtorch.utils.dataimportDataLoaderclassCustomerDataSet(Dataset):def__init__(self):super(Customer...
1train_loader = DataLoader(train_data, batch_size=4, shuffle=True)#使用数据迭代器2forid, datainenumerate(train_loader):3x_data, lable =data4print('batch:{0}, x_data:{1}, lable:{2}'.format(id, x_data, lable))