test=develop fix generate_proposals's error (#25227)

c2e07258 · FDInSky · GitHub · 36abeff4 · c2e07258
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

paddle/fluid/operators/detection/generate_proposals_op.cc paddle/fluid/operators/detection/generate_proposals_op.cc +1 -1

未找到文件。
--- 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);