Pytorch DataLoader 输出不可迭代

编程问答 2022-04-27

问题描述

我正在学习一些关于 Pytorch 的在线教程，但似乎在某些时候我没有得到预期的输出，尽管我一步一步地遵循所有内容。

import torch
import torchvision
from torchvision import datasets,transforms
Train = datasets.MNIST("",train=True,download=True,transform=transforms.Compose([transforms.ToTensor()]))
Test = datasets.MNIST("",train=False,transform=transforms.Compose([transforms.ToTensor()]))
trainset = torch.utils.data.DataLoader(Train,batch_size= 10,shuffle=True)
testset = torch.utils.data.DataLoader(Test,shuffle=True)

但是当我尝试遍历 trainset 时，我收到以下错误：

for ii in trainset:
    print(ii)
    break

TypeError: __array__() takes 1 positional argument but 2 were given

谁能告诉我为什么会出现此错误以及如何修复它？

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）

pythonpytorchtorchtorchtorch