diff --git a/src/operator/nn/concat.cc b/src/operator/nn/concat.cc index 5a7ece1459bc..2328e0a950ff 100644 --- a/src/operator/nn/concat.cc +++ b/src/operator/nn/concat.cc @@ -326,6 +326,7 @@ DMLC_REGISTER_PARAMETER(ConcatParam); [](const NodeAttrs& attrs) { \ const ConcatParam& params = nnvm::get(attrs.parsed); \ std::vector ret; \ + ret.reserve(params.num_args); \ for (int i = 0; i < params.num_args; ++i) { \ ret.push_back(std::string("arg") + std::to_string(i)); \ } \