TypeError：练习基本的pytorch数据加载器时，遇到object（）没有参数

Question

尊敬的所有亲们：

我在https://pythonprogramming.net/training-deep-learning-neural-network-pytorch/?completed=/building-deep-learning-neural-network-pytorch/的链接中尝试了该教程，并使用了Google colab笔记本上的代码。

导入和数据加载MNIST数据集的部分很好。但是，当我尝试遍历数据集时，它抱怨图片中显示了错误。我不明白为什么会这样。

Answer 1

transforms.ToTensor是一个类，您必须实例化它。

前两行应为（带有显式导入）：

import torchvision

train = torchvision.datasets.MNIST(
    "",
    train=True,
    download=True,
    # See below class is instantiated
    transform=torchvision.transforms.Compose([torchvision.transforms.ToTensor()]),
)


test = torchvision.datasets.MNIST(
    "",
    train=False,
    download=True,
    transform=torchvision.transforms.Compose([torchvision.transforms.ToTensor()]),
)

顺便说一句。如果您只有一个转换，则不需要torchvision.transforms.Compose。