1. 程式人生 > >Caffe學習:使用pycaffe進行網路訓練與測試

Caffe學習:使用pycaffe進行網路訓練與測試

使用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
    : "mnist/lenet"
  • 設定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()

即可實現訓練與測試。