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
}
}