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

[Refactor] Refactor the accumulate gradient implemention of OptimWrapper (#284)

* merge context

* update unit test

* add docstring

* fix bug in AmpOptimWrapper

* add docstring for backward

* add warning and docstring for accumuate gradient

* fix docstring

* fix docstring

* add params_group method

* fix as comment

* fix as comment

* make default_value of loss_scale to dynamic

* Fix docstring

* decouple should update and should no sync

* rename attribute in OptimWrapper

* fix docstring

* fix comment

* fix comment

* fix as comment

* fix as comment and add unit test
parent fd295741
No related branches found
No related tags found
Loading
Showing with 433 additions and 355 deletions
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment