您的位置:首页 > Web前端

caffe 保存训练的loss

2018-01-22 09:42 441 查看
命令为

./build/tools/caffe train --solver examples/DPPN/solver_dct_pixel.prototxt 2>&1| tee mylog/caffe.log

首先cd进入caffe根目录

执行该命令,

examples/DPPN/solver_dct_pixel.prototxt 为执行文件路径

mylog/caffe.log为log存放路径

如果在训练时候需要拷贝权重,则命令如下

./build/tools/caffe train --solver examples/DPPN/solver_dct_pixel.prototxt -weights=examples/DPPN/pre_train.caffemodel -gpu=0 2>&1| tee mylog/log.txt


其中,-gpu 0指定了使用gpu的id,

-weigths指定了需要拷贝的预训练模型的路径,
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: