Nigel
/
jittor
forked from jittor/jittor
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Update post_step document

This commit is contained in:
Zheng-Ning Liu 2022-06-22 19:47:27 +08:00 committed by GitHub
parent 348c0050b9
commit 47d938b2f3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
python/jittor/optim.py
View File

@ -188,7 +188,7 @@ class Optimizer(object):
def step(self, loss):
self.pre_step(loss)
...
self.post_step(loss)
self.post_step()
"""
if loss is not None:
self.backward(loss, retain_graph)
@ -203,7 +203,7 @@ class Optimizer(object):
def step(self, loss):
self.pre_step(loss)
...
self.post_step(loss)
self.post_step()
"""
jt.flags.node_order = 0
self.zero_grad()