metric-learning-divide-and-conquer
metric-learning-divide-and-conquer copied to clipboard
'_SingleProcessDataLoaderIter' object has no attribute 'next'
can someone explain to me?
import numpy as np import torch import torch.nn.functional as F import matplotlib.pyplot as plt from time import time from torchvision import datasets, transforms from torch import nn, optim
trasnform = transforms.ToTensor() trainset = datasets.MNIST('./MMIST_data/', download=True, train=True, transform=transforms) trainloader = torch.utils.data.DataLoader(trainset, batch_size=64, shuffle=True)
valset = datasets.MNIST('./MNIST_data/', download=True, train=False, transform=transforms) valloader = torch.utils.data.DataLoader(valset, batch_size=64, shuffle=True)
dataiter = iter(trainloader) imagens, etiquetas = dataiter.next() plt.imshow(imagens[0].numpy().squeeze(), cmpap='gray_r');
erro:
AttributeError Traceback (most recent call last)
AttributeError: '_SingleProcessDataLoaderIter' object has no attribute 'next'