提交 44b738e8 编写于 作者: K kpy

fix grad all ci bug

上级 13e9f5b8
...@@ -277,7 +277,7 @@ class AssignWhenInsertGrad(nn.Cell): ...@@ -277,7 +277,7 @@ class AssignWhenInsertGrad(nn.Cell):
out = self.getG(out) out = self.getG(out)
return out return out
grad_all = C.GradOperation('get_all', get_all=True) grad_all = C.GradOperation(get_all=True)
class GradNet(nn.Cell): class GradNet(nn.Cell):
def __init__(self, net): def __init__(self, net):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册