未验证 提交 728e8b17 编写于 作者: P peterzhang2029 提交者: GitHub

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

Fix useGpu in HierarchicalSigmoidLayer
...@@ -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.
先完成此消息的编辑!
想要评论请 注册