diff --git a/paddle/fluid/operators/detection/distribute_fpn_proposals_op.h b/paddle/fluid/operators/detection/distribute_fpn_proposals_op.h index 7c852934b5a5abe85f9717b77501ff8bc02c9135..f63e856626d64ec13476c3f967a085624a007c3a 100644 --- a/paddle/fluid/operators/detection/distribute_fpn_proposals_op.h +++ b/paddle/fluid/operators/detection/distribute_fpn_proposals_op.h @@ -93,7 +93,7 @@ class DistributeFpnProposalsOpKernel : public framework::OpKernel { } // define the output rois // pointer which point to each level fpn rois - T* multi_fpn_rois_data[num_level]; + std::vector multi_fpn_rois_data(num_level); // lod0 which will record the offset information of each level rois std::vector> multi_fpn_rois_lod0; for (int i = 0; i < num_level; ++i) {