pytorch 加载mnist数据集报错not gzip file

Posted wzyuan

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了pytorch 加载mnist数据集报错not gzip file相关的知识,希望对你有一定的参考价值。

利用pytorch加载mnist数据集的代码如下

import torchvision
import torchvision.transforms as transforms
from torch.utils.data import DataLoader

train_data = torchvision.datasets.MNIST(
    root=./mnist/,
    train=True,                                     # this is training data
    transform=torchvision.transforms.ToTensor(),    # Converts a PIL.Image or numpy.ndarray to
                                                    # torch.FloatTensor of shape (C x H x W) and normalize in the range [0.0, 1.0]
    download=True,
)
test_data = torchvision.datasets.MNIST(
    root=./mnist/,
    train=False,                                     # this is training data
    transform=torchvision.transforms.ToTensor(),    # Converts a PIL.Image or numpy.ndarray to
                                                    # torch.FloatTensor of shape (C x H x W) and normalize in the range [0.0, 1.0]
    download=True,
)

train_data_loader = DataLoader(train_data, shuffle=True, batch_size = 100)
test_data_loader = DataLoader(test_data,shuffle=True, batch_size=100)

第一次使用mnist,需要下载,具体方法就是设置download=True,然而我运行的时候报错了。

错误:not gzip file

可是明明是.gz文件啊,查了几篇博客也没有说清楚原因的,于是自行下载了四个文件(训练集,测试集以及各自的标签),放入./mnist/raw文件夹下,运行,报错:找不到文件。

此时,系统需要找的是./mnist/process文件夹下的train.pt和test.pt文件,这应该是pytorch下载原文件后处理生成的,可是我无处下载,于是用另一台电脑下载,程序没有报错,我把生成的.pt文件拷贝过来,可以运行了

以上是关于pytorch 加载mnist数据集报错not gzip file的主要内容,如果未能解决你的问题,请参考以下文章

用PyTorch对以MNIST数据集进行卷积神经网络

[基于Pytorch的MNIST识别02]用户数据集的读取

Pytorch实现RNN网络对MNIST字体分类

[Pytorch系列-40]:卷积神经网络 - 模型的恢复/加载 - 搭建LeNet-5网络与MNIST数据集手写数字识别

[Pytorch系列-41]:卷积神经网络 - 模型参数的恢复/加载 - 搭建LeNet-5网络与MNIST数据集手写数字识别

VS2010中RDLC报表添加数据集报错啊