pytorch的MNIST数据加载出错问题

今天学习了手bp神经网络,准备使用pytorch框架来训练一个minst数据集的模型,但是在加载数据集的时候出现了问题:
*我从官网下载了minst数据集的四个数据文件并解压,之后放在MNIST的文件夹下,想直接加载不用下载,代码如下。

import numpy as np
import matplotlib.pyplot as plt
from torch import nn,optim
from torch.autograd import Variable
import torch
from torchvision import datasets,transforms
from torch.utils.data import DataLoader
#导入数据集
train_mnist=datasets.MNIST(root="./",
                     train=True,
                     transform=transforms.ToTensor(),
                     download=False)
test_mnist=datasets.MNIST(root="./",
                     train=False,
                     transform=transforms.ToTensor(),
                     download=False)

文件目录如下:
在这里插入图片描述
MNIST文件夹跟代码放在一起,运行报错
Traceback (most recent call last):
File “C:\Users\Administrator.SC-201908231808\Desktop\py程序\pytorchtest1.py”, line 12, in
download=False)
File “G:\Program Files (x86)\Microsoft Visual Studio\Shared\Python37_64\lib\site-packages\torchvision\datasets\mnist.py”, line 73, in init
raise RuntimeError(‘Dataset not found.’ +
RuntimeError: Dataset not found. You can use download=True to download it
意思就是找不到数据,但是我换了几次路径相对路径和绝对路径都用了,还是找不到,最后发现原因是pytorch框架的MNIST
数据集加载需要两个文件如下

在这里插入图片描述
在这里插入图片描述
我以为只要把数据放在文件里就行了,实际上需要在MNIST的raw放数据,在processed中放两个.pt结尾的文件,首次使用,对pytorch不了解出现的问题。

评论 6
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值