fix paddle.shape in export model (#2091)

05e4883b · Guanghua Yu · GitHub · 2bdb9a86 · 05e4883b
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

dygraph/ppdet/modeling/layers.py dygraph/ppdet/modeling/layers.py +1 -1

未找到文件。
--- a/dygraph/ppdet/modeling/layers.py
+++ b/dygraph/ppdet/modeling/layers.py
@@ -536,7 +536,7 @@ class RCNNBox(object):
        # TODO: Updata box_clip
        origin_h = paddle.unsqueeze(origin_shape[:, 0] - 1, axis=1)
        origin_w = paddle.unsqueeze(origin_shape[:, 1] - 1, axis=1)
-        zeros = paddle.zeros(origin_h.shape, 'float32')
+        zeros = paddle.zeros(paddle.shape(origin_h), 'float32')
        x1 = paddle.maximum(paddle.minimum(bbox[:, :, 0], origin_w), zeros)
        y1 = paddle.maximum(paddle.minimum(bbox[:, :, 1], origin_h), zeros)
        x2 = paddle.maximum(paddle.minimum(bbox[:, :, 2], origin_w), zeros)