提交 a183a80e 编写于 作者: Z zlx

add param_attr for img_conv_group(...)

上级 a405e602
......@@ -340,7 +340,8 @@ def img_conv_group(input,
conv_with_batchnorm=False,
conv_batchnorm_drop_rate=0,
pool_stride=1,
pool_type=None):
pool_type=None,
param_attr=None):
"""
Image Convolution Group, Used for vgg net.
......@@ -357,6 +358,7 @@ def img_conv_group(input,
:param conv_with_batchnorm:
:param pool_stride:
:param pool_type:
:param param_attr:
:return:
"""
tmp = input
......@@ -397,6 +399,7 @@ def img_conv_group(input,
padding=conv_padding[i],
filter_size=conv_filter_size[i],
num_filters=conv_num_filter[i],
param_attr = param_attr,
**extra_kwargs)
# logger.debug("tmp.num_filters = %d" % tmp.num_filters)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册