fix the bug of use sparse_remote_update with MultiGradientMachine

9 years ago · 68c1efdd9c
parent 956217887f
commit 68c1efdd9c
1 changed files with 1 additions and 1 deletions
--- a/paddle/gserver/gradientmachines/MultiGradientMachine.cpp
+++ b/paddle/gserver/gradientmachines/MultiGradientMachine.cpp
@ -518,7 +518,7 @@ void TrainerThread::computeThread() {
        backward();
        break;
      case MultiGradientMachine::TASK_COPY_IN_ARGS:
-        copyInArgs();
+        batchSize_ = copyInArgs();
        inArgsCopied_ = true;
        multiMachine_->waitForCopyInArgs();
        break;