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文件，以及路徑正確后再運行就沒問題了。

 

有時候在下載數據集的時候可能會很慢，可能是時機不對，，，換個時間地點下載也許就好了。