From 05e4883b46f71c4f546051001a150f636f61edef Mon Sep 17 00:00:00 2001 From: Guanghua Yu <742925032@qq.com> Date: Tue, 19 Jan 2021 11:18:29 +0800 Subject: [PATCH] fix paddle.shape in export model (#2091) --- dygraph/ppdet/modeling/layers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dygraph/ppdet/modeling/layers.py b/dygraph/ppdet/modeling/layers.py index 7537a0a04..4d44fad76 100644 --- 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) -- GitLab