मैं अपने डेटा सेट के माध्यम से पुनरावृति करने की कोशिश कर रहा हूं और पहला तत्व प्राप्त कर रहा हूं

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:

क्या मुझे कुछ याद आया है!?

-1
art 13 नवम्बर 2019, 16:05

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() फ़ंक्शन को याद कर रहे हैं

0
Rishi Dev 13 नवम्बर 2019, 16:13