问题描述:

众所周知国内的网络的问题,mnist(MNIST, FashionMNIST, EMNIST)数据集不搭梯子现在是不能自动下载了。

当我去网上下载好mnist数据集之后,据说要放在:
在这里插入图片描述
然后我的代码是这样的:

import torch
from torch.utils.data import DataLoader
from torch import nn, optim
from torchvision import transforms, datasets

import visdom


def main():
    mnist_train = datasets.MNIST(root='./mnist', train=True, transform=transforms.Compose([
        transforms.ToTensor()
    ]), download=False)
    mnist_train = DataLoader(mnist_train, batch_size=32, shuffle=True)

    mnist_test = datasets.MNIST(root='./mnist', train=False, transform=transforms.Compose([
        transforms.ToTensor()
    ]), download=False)
    mnist_test = DataLoader(mnist_test, batch_size=32, shuffle=True)
    x, _ = iter(mnist_train).next()
    print('x:', x.shape)


if __name__ == '__main__':
    main()

运行之后是这样的:

Dataset not found. You can use download=True to download it
在这里插入图片描述

root='./mnist这个路径改了又改也不行

最后发现是缺少俩文件:
在这里插入图片描述


解决方法:

github上一个大佬提供了EMNIST全系列的.pt文件

链接:torchvision-datasets-mnist

在这里插入图片描述
在这里插入图片描述
最后的目录(代码没改,路径还是./mnist):
在这里插入图片描述
运行完美:
在这里插入图片描述

如果大佬的链接失效了,那我给出我的(只有mnist):

链接:https://pan.baidu.com/s/1f7Ok62Gg2xcnz4ZpQiVXiw
提取码:jdzl
复制这段内容后打开百度网盘手机App,操作更方便哦–来自百度网盘超级会员V5的分享

更多推荐