diff --git a/paddle/fluid/operators/detection/generate_proposals_op.cc b/paddle/fluid/operators/detection/generate_proposals_op.cc index 663fabd3c456185d73510a0d7570d534316b38da..981a368e8564fbcd3d688bc67d2def8664bcfe8d 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 { 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);