提交 e7f14f1a 编写于 作者: S SunAhong1993

fix

上级 b780979b
...@@ -1388,13 +1388,13 @@ class OpSet9(): ...@@ -1388,13 +1388,13 @@ class OpSet9():
"y": output_name + "__mul"}, "y": output_name + "__mul"},
outputs=[output_name]) outputs=[output_name])
else: else:
if mode == 'channel' and len(shape_slope) == 1: if mode == 'channel':
slope_data = _const_weight_or_none(val_slope) slope_data = _const_weight_or_none(val_slope)
self.weights[val_slope.name] = slope_data if len(shape_slope) > 1:
self.weights[val_slope.name] = np.reshape(slope_data, shape_slope[0])
num_parameters = val_x.out_shapes[0][1] num_parameters = val_x.out_shapes[0][1]
else: else:
num_parameters = 1 num_parameters = 1
self.paddle_graph.add_layer( self.paddle_graph.add_layer(
"paddle.nn.PReLU", "paddle.nn.PReLU",
inputs={"x": val_x.name}, inputs={"x": val_x.name},
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册