Caffe 入坑 2 ： 生成 solver 配置文件

caffe 學習筆記

生成caffe的solver文件需要調用 caffe_pb2 中的類 SolverParameter：

 

from caffe.proto import caffe_pb2

# 實例化配置類
s = caffe_pb2.SolverParameter()

# 網絡
s.train_net = "train.prototxt"
s.test_net.append("test.prototxt")

# 測試間隔
s.test_interval = 100
s.test_iter.append(10)

# 最大迭代
s.max_iter = 1000

# 學習率
s.base_lr = 0.1

# 學習率衰減係數
s.weight_decay = 5e-4

# 學習率衰減策略
s.lr_policy = "step"

# 打印間隔
s.display = 10

# 保存參數間隔
s.snapshot = 10

# 保存位置
s.snapshot_prefix = "model"

# 優化策略
s.type = "SGD"

# GPU 
s.solver_mode = caffe_pb2.SolverParameter.GPU

# 生成配置文件
with open("net/s.prototxt","w") as f:
    f.write(str(s))