pytorch Fashion MNIST 数据集下载慢怎么办

最新推荐文章于 2025-09-10 00:41:48 发布

原创

最新推荐文章于 2025-09-10 00:41:48 发布 · 2.4k 阅读

7 ·

CC 4.0 BY-SA版权

文章标签：

#python

import torch  # 导入pytorch
from torchvision import datasets, transforms ## 导入数据集与数据预处理的方法
import matplotlib.pyplot as plt

# 数据预处理：标准化图像数据，使得灰度数据在-1到+1之间
transform = transforms.Compose([transforms.ToTensor(),transforms.Normalize((0.5,), (0.5,))])

# 下载Fashion-MNIST训练集数据，并构建训练集数据载入器trainloader,每次从训练集中载入64张图片，每次载入都打乱顺序
trainset = datasets.FashionMNIST('dataset/' ,  download=True, train=True, transform=transform)
trainloader = torch.utils.data.DataLoader(trainset, batch_size=64, shuffle=True)
# 下载Fashion-MNIST测试集数据，并构建测试集数据载入器trainloader,每次从测试集中载入64张图片，每次载入都打乱顺序
testset = datasets.FashionMNIST('dataset/', download=True, train=False, transform=transform)
testloader = torch.utils.data.DataLoader(testset, batch_size=64, shuffle=True)

在python里用pytorch，执行上面的代码，卡在1%左右就一直没动。

解决办法:上GITHUB用浏览器 https://github.com/zalandoresearch/fashion-mnist/tree/maste