AttributeError: объект «_MultiProcessingDataLoaderIter» не имеет атрибута «следующий» ⇐ Python

[Anonymous]

Я пытаюсь загрузить набор данных с помощью Torch Dataset и DataLoader, но получил следующую ошибку:
AttributeError: '_MultiProcessingDataLoaderIter' object has no attribute 'next'

я использую следующий код:
class WineDataset(Dataset):

def __init__(self):
# Initialize data, download, etc.
# read with numpy or pandas
xy = np.loadtxt('./data/wine.csv', delimiter=',', dtype=np.float32, skiprows=1)
self.n_samples = xy.shape[0]

# here the first column is the class label, the rest are the features
self.x_data = torch.from_numpy(xy[:, 1:]) # size [n_samples, n_features]
self.y_data = torch.from_numpy(xy[:, [0]]) # size [n_samples, 1]

# support indexing such that dataset can be used to get i-th sample
def __getitem__(self, index):
return self.x_data[index], self.y_data[index]

# we can call len(dataset) to return the size
def __len__(self):
return self.n_samples

dataset = WineDataset()

train_loader = DataLoader(dataset=dataset,
batch_size=4,
shuffle=True,
num_workers=2)

Я пытался сделать num_workers=0, но все равно получаю ту же ошибку.
Python version 3.8.9
PyTorch version 1.13.0


Подробнее здесь: https://stackoverflow.com/questions/742 ... ibute-next