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.
D

本文介绍了如何解决在使用沐神的深度学习教程时,Fashion-MNIST数据集下载速度慢的问题。通过修改d2l包中torch.py文件的相应代码,将数据集根目录改为本地路径并禁止下载,可以避免从AWS下载,提高数据加载速度。同时提供了FashionMNIST数据集的百度网盘链接供读者下载。
最低0.47元/天 解锁文章
2621

被折叠的 条评论
为什么被折叠?



