Fix bug in MultiGradientMachine.

b93e863a · dangqingqing · 633082ad · b93e863a
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

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

未找到文件。
--- a/paddle/gserver/gradientmachines/MultiGradientMachine.cpp
+++ b/paddle/gserver/gradientmachines/MultiGradientMachine.cpp
@@ -601,7 +601,7 @@ void TrainerThread::backward() {

 void TrainerThread::backwardCallback(Parameter* para) {
  // CPU parameters are merged in the end
-  if (!para->useGpu()) return;
+  if (!para->useGpu() || para->isStatic()) return;

  int paramId = para->getID();
  if (multiMachine_->getNumThreads() == 1) {