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配置的主要内容,如果未能解决你的问题,请参考以下文章
Caffe-python interface 学习|网络训练部署测试