mmcv/tests/test_runner
Ma Zerun 357b0dfb36
[WIP] continue PR #784 (#1221)
* Add gradient cumulative optimizer

fixes #190

* Update optimizer.py

* Update optimizer.py

* fix loss scale improperly in last equivalent_iter

* Add `GradientCumulativeOptimizerHook` in `__init__.py`.

* Add docstring of `GradientCumulativeOptimizerHook`.

* Add type check, BN warning and resume warning. And fix typo, lint the
code.

* Add unit test

* Update docstring example.

* Change GradientCumulativeOptimizerHook `__init__` arguments.

* Add GradientCumulativeOptimzierHook unit tests with IterBasedRunner.

* Add GradientCumulativeFp16OptimizerHook.

* Add unit tests of GradientCumulativeFp16OptimizerHook

* Use '!=' instead of '>' to determine resume

Co-authored-by: Zhiyuan Chen <this@zyc.ai>
2021-08-23 11:18:36 +08:00
..
test_basemodule.py Fix potential bugs of basemodule when record the initilization information (#1217) 2021-08-11 22:05:44 +08:00
test_checkpoint.py [Feature] Support revise_keys in load_checkpoint(). (#829) 2021-03-03 10:59:11 +08:00
test_dist_utils.py Specifying distributed training port in os.environ when training with slurm (#362) 2020-06-20 00:49:44 +08:00
test_eval_hook.py [Fix] Fix the bug that the training log and evaluating log are mixed (#1252) 2021-08-11 22:41:21 +08:00
test_fp16.py Move unit tests to specific folders (#520) 2020-08-25 20:11:56 +08:00
test_hooks.py [WIP] continue PR #784 (#1221) 2021-08-23 11:18:36 +08:00
test_optimizer.py Support to specify LR of DCN's conv_offset (#344) 2020-09-25 20:51:58 +08:00
test_runner.py [Docs] Add header for files (#1181) 2021-08-08 22:10:55 +08:00
test_utils.py [Improvement] Improve digit_version & use it for version_checking (#1185) 2021-07-23 21:03:33 +08:00