mmpretrain/tests/test_runtime
Ezra-Yu e54cfd6951
[Imporve] Using `train_step` instead of `forward` in PreciseBNHook (#964)
* fix precise BN hook when using MLU

* fix unit tests
2022-08-11 15:02:25 +08:00
..
test_eval_hook.py [Refactor] Remove deprecation. (#633) 2022-02-10 09:30:18 +08:00
test_hooks.py [Enhance] Upgrade isort pre-commit hooks. (#687) 2022-02-17 02:17:20 +08:00
test_num_class_hook.py [Enhance] Upgrade isort pre-commit hooks. (#687) 2022-02-17 02:17:20 +08:00
test_optimizer.py [Enhance] Upgrade isort pre-commit hooks. (#687) 2022-02-17 02:17:20 +08:00
test_preciseBN_hook.py [Imporve] Using `train_step` instead of `forward` in PreciseBNHook (#964) 2022-08-11 15:02:25 +08:00