caffe学习(3)------caffe训练日志

前面介绍了怎么用caffe训练自己的图片库,在命令窗口只能看到最后的结果,这对于后期的调试网络非常不利,由官网的windows版本是找不到训练的log日志文件的,反正我是没找到,如果想看训练日志怎么办,想看训练过程的accuracy和loss到底是多少怎么办?这就得自己动手了,改代码。

用vs2013打开在src文件夹里的common.cpp文件,添加头文件 #include<direct.h>

找到voidGloballnit()函数,在::google::InitGoogleLogging(*(pargv)[0]);下面一行添加如下代码:

_mkdir("./log/");

FLAGS_colorlogtostderr =true;//设置输出到屏幕的日志显示相应颜色

google::SetLogDestination(google::GLOG_FATAL,"./log/log_error_");

google::SetLogDestination(google::GLOG_ERROR,"./log/log_error_");

google::SetLogDestination(google::GLOG_WARNING,"./log/log_error_");

google::SetLogDestination(google::GLOG_INFO,"./log/log_info_");

FLAGS_max_log_size = 1024;//最大日志大小为 1024MB

FLAGS_stop_logging_if_full_disk =true;//磁盘写满时,停止日志输出

然后编译common.cpp文件,然后重新编译生成caffe.exe文件,这样最后训练调用caffe.exe后就会在run.bat同级目录下生成一个log文件,里面就是训练日志,训练的整个过程都在里面,你可以提取里面的数据,或画loss和accuracy关于迭代次数的曲线图,这样更直观,这对调试网络很有帮助。



对于ubuntu系统,在训练的时候,比如train.sh,最后加上“2>&1 | tee your_path/your_log_name.log”,训练结束便可在your_path路径下找到名字为your_log_name的训练日志。


评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值