Mnist加载失败?掌握这些技巧,马上解决!
编程之家70
更新时间:2026-04-07 22:48:56
使用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等同样适用。
本文发布于:2026-03-14,感谢您对本站的认可!
版权声明:本站内容均来自互联网,仅供演示用,请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系,我们将在24小时内删除。
本文标签:加载数据系统编程
发布评论