diff --git a/caffe2fluid/kaffe/custom_layers/normalize.py b/caffe2fluid/kaffe/custom_layers/normalize.py index 787fcc7643cc2d374fedddcd8a2d8a1c0a65238c..ec2eb147b93bfe67bfc67ad98b4b1186a39dd6c9 100644 --- a/caffe2fluid/kaffe/custom_layers/normalize.py +++ b/caffe2fluid/kaffe/custom_layers/normalize.py @@ -47,8 +47,6 @@ def normalize_layer(input, scale_attr = fluid.ParamAttr(name=param_prefix + '_scale') scale_param = fluid.layers.create_parameter( shape=shape, dtype=input.dtype, name=name, attr=scale_attr) - scale_param = fluid.layers.reshape(x=scale_param, \ - shape=[1] if channel_shared else [input_shape[0][1]]) out = fluid.layers.elementwise_mul( x=l2_norm, y=scale_param, axis=-1 if channel_shared else 1) return out diff --git a/caffe2fluid/kaffe/custom_layers/priorbox.py b/caffe2fluid/kaffe/custom_layers/priorbox.py index e3eb6407caa7660347f81452b9668028eded81d5..f1d3135a461e0b4cd066ca9f93b06b62bc10541b 100644 --- a/caffe2fluid/kaffe/custom_layers/priorbox.py +++ b/caffe2fluid/kaffe/custom_layers/priorbox.py @@ -32,7 +32,7 @@ def priorbox_layer(inputs, name, min_size, max_size=None, - aspect_ratio=None, + aspect_ratio=1.0, variance=[0.1, 0.1, 0.2, 0.2], flip=False, clip=False,