使用snapshot繼續訓練網絡
阿新 • • 發佈:2017-11-28
2-2 rst 一行 同時 nbsp alt gpu lin 恢復
註意:snapshots和weights不能同時使用
用預訓練模型進行finetune是以下命令:
./build/tools/caffe train --solver=examples/XXX/lenet_solver.prototxt -weights examples/models/finetuning.caffemodel --gpu 1
使用快照文件(.solverstate),則可以使用-snapshot參數:
./build/tools/caffe train --solver=examples/XXX/lenet_solver.prototxt -snapshot examples/xxx/snapshot/xxx_iter_5000.solverstate --gpu 1
註意:sensenet也可以這樣訓練
在用sensenet v9訓練數據時出現以下錯誤:
這是原本的shell腳本,是利用pretrain來訓練
現在我想使用快照來恢復之前的訓練,所以我把weights註釋掉了,直接增加了snapshot。
但是一直報以下的錯誤:
就是沒有solver的意思,但是我的solver文件一直在那,我並沒有動。原因是:在vim中看起來這些行是分開的,實際上他們是通過"\"連接起來的,也就是說他們是一行。但我註釋之後,
使用snapshot繼續訓練網絡