MNIST解决网上下载FashionMNIST数据集较慢的问题

文末包含FashionMNIST数据集资源

最近在学习沐神的深度学习,在Softmax的实现中,需要导入Fashion-MNIST数据集, 并设置数据迭代器的批量大小为256。

代码如下:

import torch
from IPython import display
from d2l import torch as d2l

batch_size = 256
train_iter, test_iter = d2l.load_data_fashion_mnist(batch_size)

其中d2l.load_data_fashion_mnist要求从d2l包中加载数据集,而这里的数据集默认是亚马逊AWS下载,所以下载速度较慢,如下图:

于是根据终止程序以后的提示找到d2l文件中torch.py文件位置 

提示如下:

打开以后找到第234行(下方代码块第10、12行)。

def load_data_fashion_mnist(batch_size, resize=None):
    """Download the Fashion-MNIST dataset and then load it into memory.

    Defined in :numref:`sec_fashion_mnist`"""
    trans = [transforms.ToTensor()]
    if resize:
        trans.insert(0, transforms.Resize(resize))
    trans = transforms.Compose(trans)
    mnist_train = torchvision.datasets.FashionMNIST(
        root="../data", train=True, transform=trans, download=True)
    mnist_test = torchvision.datasets.FashionMNIST(
        root="../data", train=False, transform=trans, download=True)
    return (data.DataLoader(mnist_train, batch_size, shuffle=True,
                            num_workers=get_dataloader_workers()),
            data.DataLoader(mnist_test, batch_size, shuffle=False,
                            num_workers=get_dataloader_workers()))

root="../data"改为root="./data"  # 使用项目文件同级地址下的已有的数据集

downloda=True改为download=False  # 不下载

如下:

``` 
mnist_train = torchvision.datasets.FashionMNIST(
        root="./data", train=True, transform=trans, download=False)
mnist_test = torchvision.datasets.FashionMNIST(
        root="./data", train=False, transform=trans, download=False)
```

保存之后重启Jupyter内核再次运行代码之后,没有提示下载。


 这里附上FashionMNIST数据集链接

链接:https://pan.baidu.com/s/1GyaSjSG-_NNEj1vJelJQJg 
提取码:PyFM 


Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐