第一次,调了很久.它本来已经很OK了,同时适用CPU和GPU,且可正常运行的. 为了用于性能测试,主要改了三点: 一,每一批次显示处理时间. 二,本地加载测试数据. 三,兼容LINUX和WIN 本地加载测试数据时,要注意是用将两个pt文件,放在processed目录下,raw目录不要即可. 训练数据的定义目录是在当前目录 data/MNIST/processed目录下. 我自己弄了个下载: http://u.163.com/2FUm6N1L 提取码: XJpmqUoR 只能下载20次,过了可在…