admin 管理员组

文章数量: 1184232

使用torchvision加载数据集mnist出错出错的解决方案

错误提示为： RuntimeError: Dataset not found. You can use download=True to download it

加载数据集代码为： trainset_mnist = torchvision.datasets.MNIST(root='d:/dataset/MNIST', train=True, download=False, transform=transform_train)

错误的原因是’d:/dataset/MNIST’文件夹下没有包含数据集MNIST，torchvision中已经包含了将’MNIST’加到文件夹路径中的代码。上面的代码运行会让程序去’d:/dataset/MNIST/MNIST/'路径中查找文件，所以会找不到数据集。

正确代码为： trainset_mnist = torchvision.datasets.MNIST(root='d:/dataset', train=True, download=False, transform=transform_train)

以上方法对其他数据集如CIFAR10等同样适用。

本文标签： 加载数据 系统 编程

版权声明：本文标题：Torchvision's MNIST Missing? Learn Your Data Path & Solve the Runtime Error! 内容由网友自发贡献，该文观点仅代表作者本人， 转载请联系作者并注明出处：http://www.roclinux.cn/b/1773500648a3562647.html， 本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容，一经查实，本站将立刻删除。