提交 6ed13541 编写于 作者: G guosheng

Fix useGpu in HierarchicalSigmoidLayer

上级 976f96a9
...@@ -164,7 +164,7 @@ void HierarchicalSigmoidLayer::backward(const UpdateCallback& callback) { ...@@ -164,7 +164,7 @@ void HierarchicalSigmoidLayer::backward(const UpdateCallback& callback) {
cpuBias_ = biases_grad; cpuBias_ = biases_grad;
} }
preOutput_.grad->addByBitCodeBackward(numClasses_, *cpuLabel_, *cpuBias_); preOutput_.grad->addByBitCodeBackward(numClasses_, *cpuLabel_, *cpuBias_);
if (useGpu) { if (useGpu_) {
biases_grad->copyFrom(*cpuBias_); biases_grad->copyFrom(*cpuBias_);
} else { } else {
biases_grad = cpuBias_; biases_grad = cpuBias_;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册