Commit 6fb9ad1b by xiaotong

fix a bug

parent baad6629
...@@ -55,7 +55,7 @@ void XNetClearAll() ...@@ -55,7 +55,7 @@ void XNetClearAll()
XNet::XNet() XNet::XNet()
{ {
nodes.Clear(); nodes.Clear();
isGradEfficient = false; isGradEfficient = true;
} }
/* de-constructor */ /* de-constructor */
...@@ -163,14 +163,13 @@ void XNet::Backward(XList &roots, XList &golds, LOSS_FUNCTION_NAME loss) ...@@ -163,14 +163,13 @@ void XNet::Backward(XList &roots, XList &golds, LOSS_FUNCTION_NAME loss)
BackwardNode(node, isGradEfficient); BackwardNode(node, isGradEfficient);
if(isGradEfficient){ if(isGradEfficient){
if(!XNoder::IsLeaf(node)){ XLink & outgo = node->outgo;
XLink & outgo = node->outgo; for(int i = 0; i < outgo.tailNum; i++){
for(int i = 0; i < outgo.tailNum; i++){ XTensor * parent = outgo.tails[i];
XTensor * parent = outgo.tails[i]; ClearGrad(parent);
ClearGrad(parent);
}
} }
else
if(XNoder::IsLeaf(node))
ClearGrad(node); ClearGrad(node);
} }
} }
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论