функция загрузчика данных, и функция iter очищена, но функция next () вызывает ошибки

Question

Функция загрузчика данных используется в pytorch, и функция iter очищается, но функция next () вызывает ошибки.Если вы знаете решение или причину, пожалуйста, помогите мне.

Я изменил размер партии.

import torch
import torchvision
import torchvision.transforms as transforms
from torchvision.datasets import ImageFolder

transform = transforms.Compose([transforms.Resize(224,224),
                                transforms.ToTensor(),
                                transforms.Normalize((0.5,0.5,0.5),(0.5,0.5,0.5))])

train = ImageFolder('/home/minseok/study/dog_cat/train',transform)
valid = ImageFolder('/home/minseok/study/dog_cat/valid',transform)

train_loader = torch.utils.data.DataLoader(train,batch_size=256,num_workers=8,shuffle=True)
valid_loader = torch.utils.data.DataLoader(valid,batch_size=2,num_workers=1,shuffle=True)

dataiter = iter(valid_loader)  #It's safe up to here
images, labels = dataiter.next() # error

Извините за плохое объяснение на английском языке, потому что оно не на английском языкеговоря.

Answer 1

Вы пробовали это вместо этого?

images, labels = next(dataiter)