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

 

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

 

posted @ 2020-10-31 13:11  阳光中的影子  阅读(6131)  评论(4编辑  收藏  举报