from caffe.proto import caffe_pb2

s = caffe_pb2.SolverParameter()

path='/home/xxx/data/'

solver_file=path+'solver.prototxt'     #solver文件保存位置

s.train_net = path+'train.prototxt'     # 训练配置文件

s.test_net.append(path+'val.prototxt')  # 测试配置文件

s.test_interval = 782                   # 测试间隔

s.test_iter.append(313)                 # 测试迭代次数

s.max_iter = 78200                      # 最大迭代次数

s.base_lr = 0.001                       # 基础学习率

s.momentum = 0.9                        # momentum系数

s.weight_decay = 5e-4                   # 权值衰减系数

s.lr_policy = 'step'                    # 学习率衰减方法

s.stepsize=26067                        # 此值仅对step方法有效

s.gamma = 0.1                           # 学习率衰减指数

s.display = 782                         # 屏幕日志显示间隔

s.snapshot = 7820

s.snapshot_prefix = 'shapshot'

s.type = “SGD”                          # 优化算法

s.solver_mode = caffe_pb2.SolverParameter.GPU

with open(solver_file, 'w') as f:

    f.write(str(s))