Caffe学习：使用pycaffe进行网络训练与测试

使用pycaffe进行网络训练与测试：

• 使用数据：密码：s9ol

• 编写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()

即可实现训练与测试。

---