caffe命令行训练以及测试

本文介绍如何使用Caffe框架训练MNIST数据集,并详细解释了常见问题的解决方法,如命令行操作、日志输出及图形化展示等。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

网易云课堂(程序中文件放在目录~/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的输出,修改后运行成功

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值