提交 df8a5af9 编写于 作者: H hedaoyuan

Fix outArgs_.resize()

上级 9322f8cb
...@@ -359,9 +359,7 @@ void MultiGradientMachine::getOutArgs(std::vector<Argument>* outArgs, ...@@ -359,9 +359,7 @@ void MultiGradientMachine::getOutArgs(std::vector<Argument>* outArgs,
thread->waitOutArgsReady(); thread->waitOutArgsReady();
} }
// outArgs_.size() only need to be calculated once. outArgs_.resize(threads_[threads_.size() - 1]->getOutArgs().size());
static int size = threads_[threads_.size() - 1]->getOutArgs().size();
outArgs_.resize(size);
REGISTER_TIMER("copyOutArgs"); REGISTER_TIMER("copyOutArgs");
for (size_t i = 0; i < outArgs_.size(); ++i) { for (size_t i = 0; i < outArgs_.size(); ++i) {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册