Fix outArgs_.resize()

df8a5af9 · hedaoyuan · 9322f8cb · df8a5af9
显示空白变更内容
内联并排

Showing with 1 addition and 3 deletion

paddle/gserver/gradientmachines/MultiGradientMachine.cpp paddle/gserver/gradientmachines/MultiGradientMachine.cpp +1 -3

未找到文件。
--- a/paddle/gserver/gradientmachines/MultiGradientMachine.cpp
+++ b/paddle/gserver/gradientmachines/MultiGradientMachine.cpp
@@ -359,9 +359,7 @@ void MultiGradientMachine::getOutArgs(std::vector<Argument>* outArgs,
    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");
  for (size_t i = 0; i < outArgs_.size(); ++i) {