生成solver文件
Posted yfding的blog
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了生成solver文件相关的知识，希望对你有一定的参考价值。
from caffe.proto import caffe_pb2
s = caffe_pb2.SolverParameter()
s.train_net = "train.prototxt" # 定义网络名为trai.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.snapshop = 10.
# 定义模型存放路径
s.snapshop_prefix_prefix = "model"
s.type="SGD"
s.solver_mode=caffe_pb2.SolverParameter.GPU #定义网络优化使用gpu
#生成solver文件s.prototxt
with open(\'net/s.prototxt\',\'w\') as f:
f.write(str(s))
以上是关于生成solver文件的主要内容，如果未能解决你的问题，请参考以下文章