From b69d06472c637fb570b656ebb30c231ec90c2f3b Mon Sep 17 00:00:00 2001
From: FDInSky <48318485+FDInSky@users.noreply.github.com>
Date: Thu, 2 Jul 2020 16:54:44 +0800
Subject: [PATCH] test=develop fix generate_proposals's error (#25230)

---
 paddle/fluid/operators/detection/generate_proposals_op.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/paddle/fluid/operators/detection/generate_proposals_op.cc b/paddle/fluid/operators/detection/generate_proposals_op.cc
index 663fabd3c45..981a368e856 100644
--- a/paddle/fluid/operators/detection/generate_proposals_op.cc
+++ b/paddle/fluid/operators/detection/generate_proposals_op.cc
@@ -378,7 +378,7 @@ class GenerateProposalsKernel : public framework::OpKernel<T> {
       for (int i = 0; i < num; i++) {
         lod_data[i] = tmp_lod[i];
       }
-      rpn_rois_lod->Resize({num, 1});
+      rpn_rois_lod->Resize({num});
     }
     rpn_rois->set_lod(lod);
     rpn_roi_probs->set_lod(lod);
-- 
GitLab