1. 程式人生 > >Caffe層系列:Dropout Layer

Caffe層系列:Dropout Layer

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