如何解决Pytorch DataLoader 输出不可迭代
我正在学习一些关于 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 举报，一经查实，本站将立刻删除。
