1. 程式人生 > >caffe Python API 之Dropout

caffe Python API 之Dropout

net.pool1 = caffe.layers.Pooling(net.myconv, pool=caffe.params.Pooling.MAX, kernel_size=2, stride=2)

輸出:
layer {
  name: "pool1"
  type: "Pooling"
  bottom: "myconv"
  top: "pool1"
  pooling_param {
    pool: MAX
    kernel_size: 2
    stride: 2
  }
}