1. 程式人生 > >caffe訓練與微調時的區別

caffe訓練與微調時的區別

   以AlexNet為例,當我們自己訓練網路時,我們需要將第8層的網路結改為lr_mult:1,lr_mult:2,然後再去訓練,並且我們需要修改第8層的名稱;當我們微調時,需要將這兩個引數分別乘以10,然後再去微調網路。

PS: 檢視自己訓練網路準確率的語句:./build/tools/caffe test -model=./data/gray1/train_val.prototxt -weights=./data/gray1/caffenet_train_iter_5000.caffemodel -gpu=0