提交 68c1efdd 编写于 作者: Q qiaolongfei

fix the bug of use sparse_remote_update with MultiGradientMachine

上级 95621788
...@@ -518,7 +518,7 @@ void TrainerThread::computeThread() { ...@@ -518,7 +518,7 @@ void TrainerThread::computeThread() {
backward(); backward();
break; break;
case MultiGradientMachine::TASK_COPY_IN_ARGS: case MultiGradientMachine::TASK_COPY_IN_ARGS:
copyInArgs(); batchSize_ = copyInArgs();
inArgsCopied_ = true; inArgsCopied_ = true;
multiMachine_->waitForCopyInArgs(); multiMachine_->waitForCopyInArgs();
break; break;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册