【Pytorch】加载本地FashionMNIST数据文件

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

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
评论 8
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Snychng

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值