Merge pull request #6007 from guoshengCS/fix-hsigmoid-usegpu

Fix useGpu in HierarchicalSigmoidLayer
8 years ago · 728e8b1708
parent 976f96a9a5 6ed135413a
commit 728e8b1708
1 changed files with 1 additions and 1 deletions
--- a/paddle/gserver/layers/HierarchicalSigmoidLayer.cpp
+++ b/paddle/gserver/layers/HierarchicalSigmoidLayer.cpp
@ -164,7 +164,7 @@ void HierarchicalSigmoidLayer::backward(const UpdateCallback& callback) {
      cpuBias_ = biases_grad;
    }
    preOutput_.grad->addByBitCodeBackward(numClasses_, *cpuLabel_, *cpuBias_);
-    if (useGpu) {
+    if (useGpu_) {
      biases_grad->copyFrom(*cpuBias_);
    } else {
      biases_grad = cpuBias_;