Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Commit

Permalink
point fix the vector declaration in MultiBoxDetection (#15300)
Browse files Browse the repository at this point in the history
* point fix the vector declaration in MultiBoxDetection

* Rerun ci

* run ci

* CI
  • Loading branch information
ZhennanQin authored and TaoLv committed Jun 29, 2019
1 parent ca565a0 commit 06df38c
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions src/operator/contrib/multibox_detection.cc
Original file line number Diff line number Diff line change
Expand Up @@ -98,8 +98,7 @@ inline void MultiBoxDetectionForward(const Tensor<cpu, 3, DType> &out,
const DType *p_anchor = anchors.dptr_;

const int omp_threads = mxnet::engine::OpenMP::Get()->GetRecommendedOMPThreadCount();
std::vector<DType> outputs;
outputs.reserve(num_anchors * 6);
std::vector<DType> outputs(num_anchors * 6);
for (int nbatch = 0; nbatch < num_batches; ++nbatch) {
const DType *p_cls_prob = cls_prob.dptr_ + nbatch * num_classes * num_anchors;
const DType *p_loc_pred = loc_pred.dptr_ + nbatch * num_anchors * 4;
Expand Down

0 comments on commit 06df38c

Please sign in to comment.