caffe的solver.prototxt设置

来源:互联网 发布:江苏教育网络培训平台 编辑:程序博客网 时间:2024/06/11 20:02

# train_num = 4000

# test_num = 800

#batch_size = 100

#  epoch = 30

# test_iter = test_num / batch_size

test_iter: 8

# 每全部训练一次,测试一次

# test_interval = train_num / batch_size

test_interval: 40


# The base learning rate, momentum and the weight decay of the network.
base_lr: 0.01
momentum: 0.9

weight_decay: 0.0005

# The learning rate policy
lr_policy: "inv"
gamma: 0.0001
power: 0.75
# 每训练10次,显示一次
display: 10
# max_iter = epoch * test_interval
max_iter: 1200

# 完整训练一次,存一次model

# snapshot = train_num / batch_size

snapshot: 40

# model储存路径

snapshot_prefix: "models/"
# solver mode: CPU or GPU
solver_mode: CPU


0 0