From 9a4f578606541ac062e435ef8a448fda2f11771b Mon Sep 17 00:00:00 2001 From: Renwb1991 Date: Thu, 17 Jan 2019 20:10:32 +0800 Subject: [PATCH] caffe2fluid:fix bug in scale (#1668) --- .../caffe2fluid/kaffe/paddle/network.py | 17 ++++++++++++++--- 1 file changed, 14 insertions(+), 3 deletions(-) diff --git a/fluid/PaddleCV/caffe2fluid/kaffe/paddle/network.py b/fluid/PaddleCV/caffe2fluid/kaffe/paddle/network.py index aa910797..718bd196 100644 --- a/fluid/PaddleCV/caffe2fluid/kaffe/paddle/network.py +++ b/fluid/PaddleCV/caffe2fluid/kaffe/paddle/network.py @@ -440,7 +440,8 @@ class Network(object): if need_transpose: order = range(dims) - order.remove(axis).append(axis) + order.remove(axis) + order.append(axis) input = fluid.layers.transpose( input, perm=order, @@ -525,11 +526,21 @@ class Network(object): scale_shape = input.shape[axis:axis + num_axes] param_attr = fluid.ParamAttr(name=prefix + 'scale') scale_param = fluid.layers.create_parameter( - shape=scale_shape, dtype=input.dtype, name=name, attr=param_attr) + shape=scale_shape, + dtype=input.dtype, + name=name, + attr=param_attr, + is_bias=True, + default_initializer=fluid.initializer.Constant(value=1.0)) offset_attr = fluid.ParamAttr(name=prefix + 'offset') offset_param = fluid.layers.create_parameter( - shape=scale_shape, dtype=input.dtype, name=name, attr=offset_attr) + shape=scale_shape, + dtype=input.dtype, + name=name, + attr=offset_attr, + is_bias=True, + default_initializer=fluid.initializer.Constant(value=0.0)) output = fluid.layers.elementwise_mul( input, -- GitLab