'iterator' 对象在 python 3.7 中没有属性'next'答案

【问题标题】：'iterator' object has no attribute 'next' in python 3.7'iterator' 对象在 python 3.7 中没有属性'next'
【发布时间】：2020-03-09 07:00:34
【问题描述】：

我正在尝试遍历我的数据集并获取第一个元素

    transform = transforms.Compose([transforms.ToTensor(),transforms.Normalize((0.5),(0.5)),])
    trainloader = datasets.MNIST('~/.pytorch/MNIST_data' , download=True,train=True , transform=transform)
    ds = iter(trainloader)
    img, labels = ds.next()

但它返回此错误

    AttributeError: 'iterator' object has no attribute 'next'

我也试过了

    img , labels = next(ds)

返回此错误

    StopIteration:

我错过了什么吗？

【问题讨论】：

pytorch.org/docs/stable/_modules/torchvision/datasets/… MNIST 类实现 getitem。那么为什么不直接使用trainloader[0] 来获取第一项呢。

标签： python-3.x iterator pytorch

【解决方案1】：

如果您按照https://pytorch.org/tutorials/beginner/blitz/cifar10_tutorial.html上的教程进行操作

trainset = torchvision.datasets.CIFAR10(root='./data', train=True,
                                        download=True, transform=transform)
trainloader = torch.utils.data.DataLoader(trainset, batch_size=4,
                                          shuffle=True, num_workers=2)

dataiter = iter(trainloader)
images, labels = dataiter.next()

您的数据集上缺少 DataLoader() 函数

【讨论】：