solver_builder.py 文件源码

def build_solver(solver_filename, **kwargs):
    solver = caffe_pb2.SolverParameter()
    for k, v in kwargs.iteritems():
        setattr(solver, v)
    with open(solver_filename, 'w') as f:
        f.write(text_format.MessageToString(solver))