一、问题背景
在学习MNIST数据集手写数字识别demo的时候,笔者碰到了一些问题,现记录如下:
1、必须先确保torchvision已经正确安装。如何安装torchvision?请参考PyTorch/torchvision.datasets自带常用数据集【归纳】
2、快速下载MNIST数据集请参考我之前的这篇博文,PyTorch中MNIST数据集(附datasets.MNIST离线包)下载慢/安装慢的解决方案
3、出现EOFError: Compressed file ended before the end-of-stream marker was reached是由于第一次下载速度太慢,我强行Close terminal的原因造成,该下的文件没有下载完
二、解决方案
删除未下载好的MNIST即可。
如果在torchvison的datasets的目录没有找到,应该在运行目录。