Fix bug in MultiGradientMachine.

8 years ago · b93e863a1c
parent 633082ad9d
commit b93e863a1c
1 changed files with 1 additions and 1 deletions
--- 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) {