pytorch出现TypeError: object() takes no parameters

最新推荐文章于 2024-06-03 11:36:00 发布

一剑开天门

最新推荐文章于 2024-06-03 11:36:00 发布

阅读量393

点赞数 4

分类专栏： python 文章标签： pytorch 深度学习机器学习

本文链接：https://blog.csdn.net/m0_51170452/article/details/121754857

版权

python 专栏收录该内容

2 篇文章 0 订阅

订阅专栏

原代码

fashion_train = torchvision.datasets.FashionMNIST(root='../FashionMNIST',train=True,
                                                  transform=transforms.ToTensor,download=False)
fashion_test = torchvision.datasets.FashionMNIST(root='../FashionMNIST',train=True,
                                                 download=False,transform=transforms.ToTensor)
fashion_train_iter = Data.DataLoader(fashion_train,batch_size=256,shuffle=True)
fashion_test_iter = Data.DataLoader(fashion_test,batch_size=256,shuffle=False)

这一部分出现TypeError: object() takes no parameters错，原因是transforms.ToTensor应该加括号

fashion_train = torchvision.datasets.FashionMNIST(root='../FashionMNIST',train=True,
                                                  transform=transforms.ToTensor(),download=False)
fashion_test = torchvision.datasets.FashionMNIST(root='../FashionMNIST',train=True,
                                                 download=False,transform=transforms.ToTensor())