网易云课堂(程序中文件放在目录~/Documents/caffe/projects/mnist中):https://study.163.com/course/courseLearn.htm?courseId=1003491001#/learn/video?lessonId=1004081127&courseId=1003491001
在终端中输入sudo /home/caffe/build/tools/caffe
会出现许多帮助信息
caffe后面可以跟4个命令:train、test、device_query、time
1、time:评估模型执行的时间大概是多少
命令(跑CPU):sudo /home/caffe/build/tools/caffe time -model hbk_mnist.prototxt -iterations 10
指定模型以及迭代次数
命令(跑GPU):sudo /home/caffe/build/tools/caffe time -model hbk_mnist.prototxt -iterations 10 -gpu 0
2、train:训练网络
命令:sudo /home/caffe/build/tools/caffe train -solver hbk_mnist_solver.prototxt
出现错误:Unknown bottom blob 'label' (layer 'accuracy', bottom index 1)
将最后两层,即accuracy和cost层注释掉即可(问题:不注释怎么解决?)
将所有log都输出到一个文件中(屏幕没有输出):sudo /home/caffe/build/tools/caffe train -solver hbk_mnist_solver.prototxt 2>1 | tee a.log
!!将所有log都输出到一个文件中(屏幕既有输出,文件也有输出):sudo /home/caffe/build/tools/caffe train -solver hbk_mnist_solver.prototxt 2>&1 | tee a.log
通过caffe的工具把log中有用的信息提取出来,使用caffe/tools/extra/parse_log.py
另外一个工具更加直观的,将log输出图,使用caffe/tools/extra/plot_training_log.py.example
输出帮助信息:
使用方法:sudo python /home/caffe/tools/extra/plot_training_log.py.example 4 plotlog.png a.log
生成了一张图片plotlog.png,还有a.log.test和a.log.train两个文件
也可以用parse_log.py来生成a.log.test和a.log.train两个文件,然后自己进行画图
稍微有点问题
=====================分割线4-19======================
训练时出现的问题,因为在定义网络的文件中,即hbk_mnist.prototxt文件中,在测试阶段的数据层没有label的输出,修改后运行成功