Mnist训练环境选择参数配置

问题如图片中所示:


解决办法如下:

when the CPU_ONLY option is 1 in Makefile.config, modify the gpu option to cpu in the prototxt file as below

$ vi examples/mnist/lenet_solver.prototxt
<<<<<<<<<<<<<<<<<<<< before
 24 # solver mode: CPU or GPU
 25 solver_mode: GPU
-------------------------------------------
 24 # solver mode: CPU or GPU
 25 solver_mode: CPU
>>>>>>>>>>>>>>>>>>>>> after


解决办法参考地址。

你可能感兴趣的:(Mnist训练环境选择参数配置)