Fix useGpu in HierarchicalSigmoidLayer

release/0.11.0
guosheng 8 years ago
parent 976f96a9a5
commit 6ed135413a

@ -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_;

Loading…
Cancel
Save