caffe 训练solver配置

Posted codingstorm

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了caffe 训练solver配置相关的知识,希望对你有一定的参考价值。

学习率

net: "xxx.prototxt"
test_iter: # 测试图片总数/测试批次
test_interval: 2000
test_initialization: false

base_lr: 0.01
lr_policy: "multistep" #lr_policy: "fixed"
gamma: 0.1
stepvalue: 100000
stepvalue: 140000
stepvalue: 160000
max_iter: 200000

iter_size: 2

weight_decay: 0.0005
display: 1000
momentum: 0.9

snapshot: 5000
snapshot_prefix: "./models/xxx_"
solver_mode: GPU

以上是关于caffe 训练solver配置的主要内容,如果未能解决你的问题,请参考以下文章

用Python学习Caffe5. 生成solver文件

Caffe-python interface 学习|网络训练部署测试

Caffe-python interface 学习|网络训练部署測试

Caffe中Solver.prototxt解析

caffe solver.prototxt 生成

Caffe详解Caffe的lenet_solver.prototxt