Skip to content

Commit 6ed1354

Browse files
committed
Fix useGpu in HierarchicalSigmoidLayer
1 parent 976f96a commit 6ed1354

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)