resume_training.py 文件源码

def main():
  iter_num = process_arguments(sys.argv)
  solver_state = 'models/train_iter_{}.solverstate'.format(iter_num)

  solver = caffe.SGDSolver('solver.prototxt')
  solver.solve(solver_state) # load even *.caffemodel

  solver.net.set_mode_gpu()
  solver.net.set_device(0)

  test_interval = 1000
  max_iter = 200000

  FNULL = open(os.devnull, 'w')
  for i in xrange(max_iter):
      solver.step(1)

      if i > 0 and (i % test_interval) == 0:
        subprocess.call(['python', 'test_model.py', str(i)], stderr=FNULL)