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

暂停后继续训练:
--snapshot=xxx_iter_95000.solverstate

对已有模型进行训练:
--weights=xxx_iter_95000.caffemodel

posted on 2019-07-31 12:26  JudeHey  阅读(172)  评论(0)    收藏  举报