提交 a183a80e 编写于 作者: Z zlx

add param_attr for img_conv_group(...)

上级 a405e602
...@@ -340,7 +340,8 @@ def img_conv_group(input, ...@@ -340,7 +340,8 @@ def img_conv_group(input,
conv_with_batchnorm=False, conv_with_batchnorm=False,
conv_batchnorm_drop_rate=0, conv_batchnorm_drop_rate=0,
pool_stride=1, pool_stride=1,
pool_type=None): pool_type=None,
param_attr=None):
""" """
Image Convolution Group, Used for vgg net. Image Convolution Group, Used for vgg net.
...@@ -357,6 +358,7 @@ def img_conv_group(input, ...@@ -357,6 +358,7 @@ def img_conv_group(input,
:param conv_with_batchnorm: :param conv_with_batchnorm:
:param pool_stride: :param pool_stride:
:param pool_type: :param pool_type:
:param param_attr:
:return: :return:
""" """
tmp = input tmp = input
...@@ -397,6 +399,7 @@ def img_conv_group(input, ...@@ -397,6 +399,7 @@ def img_conv_group(input,
padding=conv_padding[i], padding=conv_padding[i],
filter_size=conv_filter_size[i], filter_size=conv_filter_size[i],
num_filters=conv_num_filter[i], num_filters=conv_num_filter[i],
param_attr = param_attr,
**extra_kwargs) **extra_kwargs)
# logger.debug("tmp.num_filters = %d" % tmp.num_filters) # 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.
先完成此消息的编辑!
想要评论请 注册