@@ -9,13 +9,15 @@ def simple_img_conv_pool(input,
99 pool_size ,
1010 pool_stride ,
1111 act ,
12+ param_attr = None ,
1213 pool_type = 'max' ,
1314 main_program = None ,
1415 startup_program = None ):
1516 conv_out = layers .conv2d (
1617 input = input ,
1718 num_filters = num_filters ,
1819 filter_size = filter_size ,
20+ param_attr = param_attr ,
1921 act = act ,
2022 main_program = main_program ,
2123 startup_program = startup_program )
@@ -36,6 +38,7 @@ def img_conv_group(input,
3638 conv_padding = 1 ,
3739 conv_filter_size = 3 ,
3840 conv_act = None ,
41+ param_attr = None ,
3942 conv_with_batchnorm = False ,
4043 conv_batchnorm_drop_rate = None ,
4144 pool_stride = 1 ,
@@ -57,6 +60,7 @@ def __extend_list__(obj):
5760
5861 conv_padding = __extend_list__ (conv_padding )
5962 conv_filter_size = __extend_list__ (conv_filter_size )
63+ param_attr = __extend_list__ (param_attr )
6064 conv_with_batchnorm = __extend_list__ (conv_with_batchnorm )
6165 conv_batchnorm_drop_rate = __extend_list__ (conv_batchnorm_drop_rate )
6266
@@ -70,6 +74,7 @@ def __extend_list__(obj):
7074 num_filters = conv_num_filter [i ],
7175 filter_size = conv_filter_size [i ],
7276 padding = conv_padding [i ],
77+ param_attr = param_attr [i ],
7378 act = local_conv_act ,
7479 main_program = main_program ,
7580 startup_program = startup_program )
@@ -101,6 +106,7 @@ def __extend_list__(obj):
101106def sequence_conv_pool (input ,
102107 num_filters ,
103108 filter_size ,
109+ param_attr = None ,
104110 act = "sigmoid" ,
105111 pool_type = "max" ,
106112 main_program = None ,
@@ -109,6 +115,7 @@ def sequence_conv_pool(input,
109115 input = input ,
110116 num_filters = num_filters ,
111117 filter_size = filter_size ,
118+ param_attr = param_attr ,
112119 act = act ,
113120 main_program = main_program ,
114121 startup_program = startup_program )
0 commit comments