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

最新推荐文章于 2025-03-08 08:39:19 发布

大马鱼哈

最新推荐文章于 2025-03-08 08:39:19 发布

阅读量2.3k

点赞数 8

分类专栏： Python 文章标签： python

本文链接：https://blog.youkuaiyun.com/majian/article/details/106215965

版权

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用浏览器