源代码
train_dataset = datasets.MNIST(root='./data',train=True,transform=transforms.ToTensor(),download=False)
test_dataset = datasets.MNIST(root='./data',train=False,transform=transforms.ToTensor)
train_loader = torch.utils.data.DataLoader(dataset=train_dataset,batch_size=batch_size,shuffle=True)
test_loader =torch.utils.data.DataLoader(dataset=test_dataset,batch_size=batch_size,shuffle=True,num_workers=0)
for data, target in test_loader:
print(data)
报错代码
TypeError: __init__() takes 1 positional argument but 2 were given
问题解决
仔细检查发现test_dataset = datasets.MNIST(root='./data',train=False,transform=transforms.ToTensor)应该加个括号,改正为datasets.MNIST(root='./data',train=False,transform=transforms.ToTensor())