caffe ubuntu下ctrl+C不能保存模型

最新在ubuntu下训练caffe模型时,保存了输出日志的同时,无法使用ctrl+C保存模型,诸多不便,采用的shell命令如下:

modelroot=...
log=train-`date +%Y-%m-%d-%H-%M-%S`.log
./build/tools/caffe train -solver $modelroot/solver.prototxt 2>1 | tee $log $@

查阅相关资料,
问题1,找到问题2,进而找到问题3,
出现的问题为tee命令将ctrl+c可能做了转换处理,使得caffe命令无法接收到正确的中断命令。
解决的方法有:
方法1:
不使用tee命令来保存日志,生成的日志事后需要重新命名。

GLOG_log_dir=/path/to/log/dir $CAFFE_ROOT/bin/caffe.bin train 
—solver=/path/to/solver.prototxt

方法2:
继续使用tee命令保存日志,使用kill命令来给caffe命令发送中断命令。

kill -s SIGINT <proc_id>

倾向于第二种方法,亲测有效。

同时,在训练时加入-SIGINT_effect snapshot导致使用ctrl+c时,caffe命令不会中断,会保存模型,之后继续训练,需要使用kill proc_id来强行关闭进程。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值