From 507f47973270156095726506e9a1cf68f9fb4b05 Mon Sep 17 00:00:00 2001 From: minqiyang Date: Wed, 15 Aug 2018 19:06:43 +0800 Subject: [PATCH] Polish code --- python/paddle/fluid/layers/detection.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/paddle/fluid/layers/detection.py b/python/paddle/fluid/layers/detection.py index abe479693..9baf5f84f 100644 --- a/python/paddle/fluid/layers/detection.py +++ b/python/paddle/fluid/layers/detection.py @@ -1111,7 +1111,7 @@ def multi_box_head(inputs, mbox_loc = nn.transpose(mbox_loc, perm=[0, 2, 3, 1]) compile_shape = [ mbox_loc.shape[0], cpt.floor_division( - box_loc.shape[1] * mbox_loc.shape[2] * mbox_loc.shape[3], 4), 4 + mbox_loc.shape[1] * mbox_loc.shape[2] * mbox_loc.shape[3], 4), 4 ] run_shape = tensor.assign(numpy.array([0, -1, 4]).astype("int32")) mbox_loc_flatten = nn.reshape( -- GitLab