Skip to content

Commit 728e8b1

Browse files
Merge pull request #6007 from guoshengCS/fix-hsigmoid-usegpu
Fix useGpu in HierarchicalSigmoidLayer
2 parents 976f96a + 6ed1354 commit 728e8b1

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

paddle/gserver/layers/HierarchicalSigmoidLayer.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -164,7 +164,7 @@ void HierarchicalSigmoidLayer::backward(const UpdateCallback& callback) {
164164
cpuBias_ = biases_grad;
165165
}
166166
preOutput_.grad->addByBitCodeBackward(numClasses_, *cpuLabel_, *cpuBias_);
167-
if (useGpu) {
167+
if (useGpu_) {
168168
biases_grad->copyFrom(*cpuBias_);
169169
} else {
170170
biases_grad = cpuBias_;

0 commit comments

Comments
 (0)