caffe继续训练

只需要在开始执行训练的脚本上,添加参数–snapshot即可,因为之前训练的过程中会生成两个文件一个是.caffemodel,另一个是.solverstate文件,我们一般是使用.caffemodel去进行预测的,而.solverstate文件方便我们继续训练的。

GLOG_logtostderr=0 GLOG_log_dir=E:\BaiduNetdisk\caffe-windows\examples\pixel\5s\logs\
E:\BaiduNetdisk\caffe-windows\Build\x64\Release\caffe.exe train --solver=pixel_solver.prototxt --snapshot=model/_iter_1000000.solverstate
pause

猜你喜欢

转载自blog.csdn.net/czp_374/article/details/85079266