!5262 fix grad all ci bug
Merge pull request !5262 from flywind/fix_grad_all
This commit is contained in:
commit
c0e02bd88b
|
@ -277,7 +277,7 @@ class AssignWhenInsertGrad(nn.Cell):
|
|||
out = self.getG(out)
|
||||
return out
|
||||
|
||||
grad_all = C.GradOperation('get_all', get_all=True)
|
||||
grad_all = C.GradOperation(get_all=True)
|
||||
|
||||
class GradNet(nn.Cell):
|
||||
def __init__(self, net):
|
||||
|
|
Loading…
Reference in New Issue