Open LUCASDNORONHA opened 1 year ago
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) in 1 dataiter = iter(trainloader) ----> 2 imagens, etiquetas = dataiter.next() 3 plt.imshow(imagens[0].numpy().squeeze(), cmpap='gray_r');
AttributeError: '_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) in
1 dataiter = iter(trainloader)
----> 2 imagens, etiquetas = dataiter.next()
3 plt.imshow(imagens[0].numpy().squeeze(), cmpap='gray_r');
AttributeError: '_SingleProcessDataLoaderIter' object has no attribute 'next'