看起来你已经做对了。 该 weight_decay 元参数结合 regularization_type: "L2" 在你的 'solver.prototxt' 告诉caffe使用 L2 正规化 weight_decay = 1e-5 。
weight_decay
regularization_type: "L2"
'solver.prototxt'
L2
weight_decay = 1e-5
您可能想要调整的另一件事是正则化对每个参数的影响程度。你可以设置它 对于网络中的每个参数blob 通过
param { decay_mult: 1 }
例如,一个 "InnerProduct" 带偏置的图层有两个参数:
"InnerProduct"
layer { type: "InnerProduct" name: "fc1" # bottom and top here inner_product_param { bias_term: true # ... other params } param { decay_mult: 1 } # for weights use regularization param { decay_mult: 0 } # do not regularize the bias }
默认情况下, decay_mult 设置为1,也就是说,网的所有权重都是相同的。您可以更改它以规范更多/更少特定的参数blob。
decay_mult