Skip to content

Commit b69d064

Browse files
authored
test=develop fix generate_proposals's error (#25230)
1 parent fcd4cf7 commit b69d064

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

paddle/fluid/operators/detection/generate_proposals_op.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -378,7 +378,7 @@ class GenerateProposalsKernel : public framework::OpKernel<T> {
378378
for (int i = 0; i < num; i++) {
379379
lod_data[i] = tmp_lod[i];
380380
}
381-
rpn_rois_lod->Resize({num, 1});
381+
rpn_rois_lod->Resize({num});
382382
}
383383
rpn_rois->set_lod(lod);
384384
rpn_roi_probs->set_lod(lod);

0 commit comments

Comments
 (0)