如何在卷积层pycaffe中设置bias_term false?

时间:2016-12-18 21:50:05

标签: python deep-learning caffe pycaffe

我正在使用pycaffe来创建我的网络,我想将卷积层中的偏置项设置为false,但是找不到任何操作方法。到目前为止我的代码片段:

import caffe
from caffe import layers als L, params as P

n.conv1 = L.Convolution(n.data,kernel_size = 3,stride = 1,num_output=16,pad=1,weight_filler=dict(type='xavier'))

1 个答案:

答案 0 :(得分:2)

正如您已经提到的,这是通过将bias_term参数设置为false来完成的。 通常,您可以在Layer Catalogue中找到大多数图层及其参数。您只需使用Layer目录中记录的名称和值,即可从PyCaffe设置任何参数。请记住,您必须使用正确的Python语法,即False而不是false

n.conv1 = L.Convolution(n.data,
                        kernel_size=3,
                        stride=1,
                        num_output=16,
                        pad=1,
                        weight_filler=dict(type='xavier'),
                        bias_term=False
                        )

这将在.prototxt文件中创建以下条目:

layer {
  name: "conv1"
  type: "Convolution"
  bottom: "data"
  top: "conv1"
  convolution_param {
    num_output: 16
    bias_term: false
    pad: 1
    kernel_size: 3
    stride: 1
    weight_filler {
      type: "xavier"
    }
  }
}

如您所见,该选项已被正确识别并放入convolution_param块内。