TypeError: ToTensor() takes no arguments
编写encoder时遇到的问题。
mnist_test = datasets.MNIST(root='./dataset/mnist',train=False,
transform=transforms.Compose([transforms.ToTensor]),
download=True)
出现TypeError: ToTensor() takes no arguments,
原因是ToTensor后忘记加括号(),正确的应该为:
mnist_test = datasets.MNIST(root='./dataset/mnist',train=False,
transform=transforms.Compose([transforms.ToTensor()]),
download=True)