我正在使用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'))
答案 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
块内。