1. 程式人生 > >使用snapshot繼續訓練網絡

使用snapshot繼續訓練網絡

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繼續訓練網絡