From a183a80e2e41181272c72dd2afb6eb13aa9a31b7 Mon Sep 17 00:00:00 2001 From: zlx Date: Fri, 21 Jul 2017 16:14:36 +0800 Subject: [PATCH] add param_attr for img_conv_group(...) --- python/paddle/trainer_config_helpers/networks.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/python/paddle/trainer_config_helpers/networks.py b/python/paddle/trainer_config_helpers/networks.py index 1bf59ed48..789d4f170 100755 --- a/python/paddle/trainer_config_helpers/networks.py +++ b/python/paddle/trainer_config_helpers/networks.py @@ -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) -- GitLab