如何在无网络环境下下载并使用torchvision数据集?
离线运行FashionMNIST示例教程
操作步骤
- 第一步:先在有网络的设备上运行一次你给出的原始代码,运行完成后,会在代码所在目录生成
data/FashionMNIST/raw文件夹,里面存储了FashionMNIST的全部4个原始数据文件(可能带.gz压缩后缀,无需手动解压)。 - 第二步:将整个
FashionMNIST文件夹拷贝到无网络设备的目标路径下,保持路径结构为[你的项目目录]/data/FashionMNIST/raw/[4个数据文件]。注意不要修改raw文件夹内的文件名,否则torchvision会识别失败。 - 第三步:修改原代码中的
download参数,将值从True改为False,修改后的完整代码如下:
import torch from torch import nn from torch.utils.data import DataLoader from torchvision import datasets from torchvision.transforms import ToTensor, Lambda training_data = datasets.FashionMNIST( root="data", train=True, download=False, transform=ToTensor() ) test_data = datasets.FashionMNIST( root="data", train=False, download=False, transform=ToTensor() )
常见问题说明
- 如果你的数据集没有放在项目下的
data目录,直接修改root参数的值为你实际存放FashionMNIST文件夹的父目录路径即可。 - 只要路径和文件名正确,离线环境下代码可以直接运行,不需要其他额外配置。
内容的提问来源于stack exchange,提问作者John Smith
相关产品推荐
相关产品推荐

