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文件,以及路径正确后再运行就没问题了。

 

有时候在下载数据集的时候可能会很慢,可能是时机不对,,,换个时间地点下载也许就好了。

 


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM