# -*- coding: utf-8 -*-
"""
Created on Sun Jul 17 18:20:57 2016
@author: root
"""
path='/home/xxx/data/'
solver_file=path+'solver.prototxt'     #solver文件保存位置
sp={}
sp['train_net']=‘“'+path+'train.prototxt”'  # 訓(xùn)練配置文件
sp['test_net']=‘“'+path+'val.prototxt”'     # 測試配置文件
sp['test_iter']='313'                  # 測試迭代次數(shù)
sp['test_interval']='782'              # 測試間隔
sp['base_lr']='0.001'                  # 基礎(chǔ)學(xué)習(xí)率
sp['display']='782'                    # 屏幕日志顯示間隔
sp['max_iter']='78200'                 # 最大迭代次數(shù)
sp['lr_policy']='“step”'                 # 學(xué)習(xí)率變化規(guī)律
sp['gamma']='0.1'                      # 學(xué)習(xí)率變化指數(shù)
sp['momentum']='0.9'                   # 動量
sp['weight_decay']='0.0005'            # 權(quán)值衰減
sp['stepsize']='26067'                 # 學(xué)習(xí)率變化頻率
sp['snapshot']='7820'                   # 保存model間隔
sp['snapshot_prefix']=‘"snapshot"'       # 保存的model前綴
sp['solver_mode']='GPU'                # 是否使用gpu
sp['solver_type']='SGD'                # 優(yōu)化算法
def write_solver():
    #寫入文件
    with open(solver_file, 'w') as f:
        for key, value in sorted(sp.items()):
            if not(type(value) is str):
                raise TypeError('All solver parameters must be strings')
            f.write('%s: %s\n' % (key, value))
if __name__ == '__main__':
    write_solver()

# -*- coding: utf-8 -*-
from caffe.proto import caffe_pb2
s = caffe_pb2.SolverParameter()
path='/home/xxx/data/'
solver_file=path+'solver1.prototxt'
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
s.weight_decay = 5e-4
s.lr_policy = 'step'
s.stepsize=26067
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))

import caffe
caffe.set_device(0)
caffe.set_mode_gpu()
solver = caffe.SGDSolver('/home/xxx/data/solver.prototxt')
solver.solve()