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_base_model

[Fix] Fix image dtype when enable_normalize=False. (#301)

2022-06-13 21:21:19 +08:00

test_wrappers

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

2022-06-13 23:20:53 +08:00

test_averaged_model.py

[Fix] Fix ema state dict swapping in EMAHook and torch1.5 ut. (#266)

2022-05-30 16:51:06 +08:00

test_base_module.py

[Feature] Add ModuleList Sequential and ModuleDict (#299)

2022-06-13 13:51:07 +08:00