Caffe學習:使用pycaffe進行網路訓練與測試
阿新 • • 發佈:2019-02-09
使用pycaffe進行網路訓練與測試:
編寫solver檔案:
train_net: "mnist/lenet_auto_train.prototxt" test_net: "mnist/lenet_auto_test.prototxt" test_iter: 100 # 每一次測試時進行100次forward pass test_interval: 500 base_lr: 0.01 momentum: 0.9 weight_decay: 0.0005 lr_policy: "inv" gamma: 0.0001 power: 0.75 display: 100 max_iter: 10000 snapshot: 5000 snapshot_prefix
設定Caffe使用GPU模式:
import caffe caffe.set_device(0) caffe.set_mode_gpu()
根據solver檔案建立solver(使用SGDSolver):
solver = caffe.SGDSolver('mnist/lenet_auto_solver.prototxt')
開始進行解算:
solver.solve()
即可實現訓練與測試。