如何根据需要在Caffe中进行网络拍摄？

Question

我希望能够在培训进行时按需提供网络快照（比如在某些条件下）。有没有办法用Caffe做到这一点？

Answer 1

例如Python中的回调：

import caffe

def OnStart():
  pass # both callbacks must be defined anyway 

def OnGradientsReady():
  global solver
  if solver.iter == 17:
    solver.snapshot()

solver = caffe.get_solver("mnist/lenet_solver_t1.prototxt")
solver.add_callback(OnStart, OnGradientsReady)
solver.solve()