项目场景:
在李沐老师视频课Softmax 回归 + 损失函数 + 图片分类数据集【动手学深度学习v2】中用到了FashionMNIST数据集。
问题描述:
trans = transforms.ToTensor()
#下载到上一级目录的data文件中,下载的是训练数据集,类型时tensor,从网上下载
mnist_train = torchvision.datasets.FashionMNIST(
root="../data", train=True, transform=trans, download=True)
mnist_test = torchvision.datasets.FashionMNIST(
root="../data", train=False, transform=trans, download=True)
从网上下载数据集失败,[WinError 10054] 远程主机强迫关闭了一个现有的连接
原因分析:
由于各种原因我们很难从国外网站上下载数据集
解决方案:
1.在网上下载FashionMNIST数据集-github
2.在上一级的data目录下新建一个FashionMNIST
目录,在这个目录下新建两个目录processed
和raw
3.将下载到的数据集中四个压缩包移到raw
文件夹中并解压
4.再执行一下代码
trans = transforms.ToTensor()
#下载到上一级目录的data文件中,下载的是训练数据集,类型时tensor,从网上下载
mnist_train = torchvision.datasets.FashionMNIST(
root="../data", train=True, transform=trans, download=True)
mnist_test = torchvision.datasets.FashionMNIST(
root="../data", train=False, transform=trans, download=True)
试验一下是否读取成功
len(mnist_train), len(mnist_test)
(60000, 10000)