Skip to content
Snippets Groups Projects
Unverified Commit 58678234 authored by Mashiro's avatar Mashiro Committed by GitHub
Browse files

[Enhance] Support pass arguments to `OptimWrapper.step/backward/zero_grad` (#472) 

* add args to OptimWrapper.step backward zero_grad

* minor refine

* minor refine
parent 8d25dbde
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment

Datenschutz und Impressum