【caffe使用记录】caffe的train和test网络model

来源:互联网 发布:火龙果软件培训 编辑:程序博客网 时间:2024/06/15 18:49

无初始化训练,即从0开始训练:

#!/user/bin/env bashset -ecafferoot/bulid/tools/caffe train --solver=path/to/solver.prototxt


用现有的模型初始化训练:

#!/user/bin/env bashset -ecafferoot/bulid/tools/caffe train --solver=path/to/solver.prototxt --weight=path/to/model.caffemodel


 我们在使用caffe训练过程中会生成.caffemodel和.solverstate文件,一个是模型文件,一个是中间状态文件(生成多少个取决于你自己设定的snapshot)。当训练过程中断,你想继续运行数据学习,此时只需要调用.solverstate文件即可。

#!/user/bin/env bsahset -ecafferoot/build/tools/caffe train   --solver=path/to/solver.prototxt --snapshot=path/to/test_100000.solverstate  


模型训练后忘记模型的精度:

cafferoot/build/tools/caffe test -model=path/to/train_test.prototxt -weights=path/to/lenet_iter_10000.caffemodel -gpu=0 





0 0
原创粉丝点击