Mnist训练环境选择参数配置

来源:互联网 发布:机顶盒电影软件 编辑:程序博客网 时间:2024/06/04 01:08

问题如图片中所示:


解决办法如下:

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


解决办法参考地址。

0 0
原创粉丝点击