论坛 › Python交流 › 请问有谁知道pytorch的MNIST数据集出现下列报错怎么处理，源码和报错截图如下

遥远的拟 发表于 2021-9-25 10:59:34

请问有谁知道pytorch的MNIST数据集出现下列报错怎么处理，源码和报错截图如下

train_loader = torch.utils.data.DataLoader(
torchvision.datasets.MNIST('./data/', train=True, download=True,
                           transform=torchvision.transforms.Compose([
                               torchvision.transforms.ToTensor(),
                               torchvision.transforms.Normalize(
                                 (0.1307,), (0.3081,))
                           ])),
batch_size=batch_size_train, shuffle=True)
test_loader = torch.utils.data.DataLoader(
torchvision.datasets.MNIST('./data/', train=False, download=True,
                           transform=torchvision.transforms.Compose([
                               torchvision.transforms.ToTensor(),
                               torchvision.transforms.Normalize(
                                 (0.1307,), (0.3081,))
                           ])),
batch_size=batch_size_test, shuffle=True)
​

AdiosSpike 发表于 2021-9-25 15:35:41

谢谢

查看完整版本: 请问有谁知道pytorch的MNIST数据集出现下列报错怎么处理，源码和报错截图如下