版权声明:未经博主允许,不得转载! https://blog.csdn.net/u011681952/article/details/86157124
Dropout Layer作用是随机让网络的某些节点不工作(输出置零),也不更新权重;是防止模型过拟合的一种有效方法
首先我们先看一下 DropoutParameter
message DropoutParameter {
optional float dropout_ratio = 1 [default = 0.5]; // dropout ratio
}
InnerProduct layer 在prototxt里面的书写:
layer {
name: "drop"
type: "Dropout"
bottom: "fc"
top: "fc"
dropout_param {
dropout_ratio: 0.5
}
}