solver.txt 示例:
对于SGD,是不需要 "momentum2" "delta" "type" 参数的。
net : "xxx.prototxt"
# The base learning rate, momentum and the weight decay of the network.
base_lr: 0.00
momentum: 0.9
momentum2:0.999
delta:1e-8
weight_decay: 0.0 #lambda
# The learning rate policy
lr_policy: "step"
stepsize: 30000 #the base_lr = base_lr * gamma^step
gamma: 0.8
display: 500
clip_gradients:10
max_iter: 600000
snapshot: 2500
snapshot_prefix: "./models/"
type:"Adam"
solver_mode: GPU
本文介绍了一个用于训练神经网络的solver配置文件示例,重点讲解了SGD与Adam两种优化器的不同参数设置,包括学习率策略、动量项等。

426

被折叠的 条评论
为什么被折叠?



