码迷,mamicode.com
首页 > 其他好文 > 详细

caffe 训练solver配置

时间:2019-07-31 13:02:03      阅读:97      评论:0      收藏:0      [点我收藏+]

标签:snapshot   AMM   init   false   总数   ref   prot   models   moment   

学习率

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配置

标签:snapshot   AMM   init   false   总数   ref   prot   models   moment   

原文地址:https://www.cnblogs.com/codingstorm/p/11275563.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!