目前还不太清楚为何训练中会出现 Accuracy 突降、Loss 突增的问题。 本项目测试环境为 Ubuntu20.04，python 版本为 3.10.13。 根据自己的需要修改 train.py 文件中第 11~23 行的参数（默认也可以）。 运行 train.py 即可：python train.py。 该项目每轮训练中，只要训练准确率比 ...

在MNIST上进行训练，可以说是计算机视觉里的“Hello World”任务了。而如果使用PyTorch的标准代码训练CNN，一般需要3分钟左右。 本文经AI新媒体量子位（公众号ID:QbitAI）授权转载，转载请联系出处。 在MNIST上进行训练，可以说是计算机视觉里的“Hello World”任务了。