Caffe——python接口学习（2）：生成solver文件

caffe在训练的时候，需要一些参数设置，一般将这个参数设置在一个叫solver.prototxt的文件里面，如下：

base_lr = 0.001
display:782
gamma : 0.1
lr_policy : "step"
max_iter :78200
monentum: 0.9
snapshot:7820
snapshot_prefix:"snapshot"
solver_mode:GPU
solver_type:SGD
stepsize:26067
test_interval:782
test_iter:313
test_net:"/home/liu/data/val.prototxt"
train_net:"/home/liu/data/proto/train.prototxt"
weight_decay:0.0005

有一些参数需要计算，不能呢乱设置。
假设我们有50000个训练样本，batch_size= 64,即每批处理64个样本，那么需要50000/64=782次才能处理完一次全部样本。我们把处理完一次所有样本称之为一代，即epoch。所以，这里的test_interval设置为782，即处理完一次所有训练样本后才进行测试。如果我们想训练100代，则需要设置max_iter为78200。
同理，如果有10000个测试样本，batch_size= 32，那么需要迭代10000/32 = 313次才训练完，所以设置test_iter为313
学习率变换规律我们设置随着迭代次数增加，慢慢变低。总共迭代78200次，我们将变化lr_rate三次，所以stepsize设置为78200/3 = 26067，即每迭代26067次，我们就降低一次学习率。
我们可以用python实现：

# -*- 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”'  # 训练配置文件
sp['test_net']=‘“’+path+'val.prototxt”'     # 测试配置文件
sp['test_iter']='313'                  # 测试迭代次数
sp['test_interval']='782'              # 测试间隔
sp['base_lr']='0.001'                  # 基础学习率
sp['display']='782'                    # 屏幕日志显示间隔
sp['max_iter']='78200'                 # 最大迭代次数
sp['lr_policy']='“step”'                 # 学习率变化规律
sp['gamma']='0.1'                      # 学习率变化指数
sp['momentum']='0.9'                   # 动量
sp['weight_decay']='0.0005'            # 权值衰减
sp['stepsize']='26067'                 # 学习率变化频率
sp['snapshot']='7820'                   # 保存model间隔
sp['snapshot_prefix']=‘"snapshot"’       # 保存的model前缀
sp['solver_mode']='GPU'                # 是否使用gpu
sp['solver_type']='SGD'                # 优化算法

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()