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

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

巴特西

pytorch MNIST加载已下载的数据集出现问题及解决方法

最新文章

热门文章