mmengine/tests/test_optim
Mashiro b7866021c4
[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
2022-06-13 23:20:53 +08:00
..
test_optimizer [Refactor] Refactor the accumulate gradient implemention of OptimWrapper (#284) 2022-06-13 23:20:53 +08:00
test_scheduler [Features]Add OneCycleLR (#296) 2022-06-13 21:23:59 +08:00