pytorch MNIST加载已下载的数据集出现问题及解决方法
import torch import torch.nn as nn from torch.autograd import Variable import torchvision import torch.utils.data as Data import matplotlib.pyplot as plt DOWNLOAD_MNIST=False train_data=torchvision.datasets.MNIST( root='./mnist/MNIST/raw', # train=True, transform=torchvision.transforms.ToTensor(), download=DOWNLOAD_MNIST, )
用以上代码在加载自己的数据集的时候出现以下问题:
raise RuntimeError('Dataset not found.' +
RuntimeError: Dataset not found. You can use download=True to download it
原因及解决方法:
我的py文件是和mnist文件夹在同一个目录下的,但是根据我在代码中给出的目录没有能找到我的数据集,百度了一下,最终发现是一下原因:
路径只要设置到MNIST上一级目录就好,针对我给出的上面的图,相对路径应该设置为root='./minst'。
以上是第一个问题,但是解决之后发现还是出现最初的错误,原因是只下载了4个.gz文件,而缺少了在processed文件夹里的两个.pt文件。
当有了这两个.pt文件,以及路径正确后再运行就没问题了。
有时候在下载数据集的时候可能会很慢,可能是时机不对,,,换个时间地点下载也许就好了。