[Refactor] Add batch_idx to hook input. (#140)
* [Refactor] Add batch_idx to hook input. * update
Showing
- mmengine/hooks/checkpoint_hook.py 9 additions, 7 deletionsmmengine/hooks/checkpoint_hook.py
- mmengine/hooks/empty_cache_hook.py 2 additions, 0 deletionsmmengine/hooks/empty_cache_hook.py
- mmengine/hooks/hook.py 53 additions, 15 deletionsmmengine/hooks/hook.py
- mmengine/hooks/iter_timer_hook.py 8 additions, 5 deletionsmmengine/hooks/iter_timer_hook.py
- mmengine/hooks/logger_hook.py 10 additions, 5 deletionsmmengine/hooks/logger_hook.py
- mmengine/hooks/naive_visualization_hook.py 2 additions, 0 deletionsmmengine/hooks/naive_visualization_hook.py
- mmengine/hooks/optimizer_hook.py 3 additions, 1 deletionmmengine/hooks/optimizer_hook.py
- mmengine/hooks/param_scheduler_hook.py 3 additions, 1 deletionmmengine/hooks/param_scheduler_hook.py
- mmengine/runner/loops.py 20 additions, 14 deletionsmmengine/runner/loops.py
- mmengine/runner/runner.py 0 additions, 6 deletionsmmengine/runner/runner.py
- tests/test_hook/test_checkpoint_hook.py 4 additions, 3 deletionstests/test_hook/test_checkpoint_hook.py
- tests/test_hook/test_empty_cache_hook.py 5 additions, 5 deletionstests/test_hook/test_empty_cache_hook.py
- tests/test_hook/test_hook.py 5 additions, 7 deletionstests/test_hook/test_hook.py
- tests/test_hook/test_iter_timer_hook.py 16 additions, 16 deletionstests/test_hook/test_iter_timer_hook.py
- tests/test_hook/test_logger_hook.py 13 additions, 10 deletionstests/test_hook/test_logger_hook.py
- tests/test_hook/test_naive_visualization_hook.py 8 additions, 7 deletionstests/test_hook/test_naive_visualization_hook.py
- tests/test_hook/test_optimizer_hook.py 3 additions, 3 deletionstests/test_hook/test_optimizer_hook.py
- tests/test_hook/test_param_scheduler_hook.py 8 additions, 8 deletionstests/test_hook/test_param_scheduler_hook.py
- tests/test_hook/test_sync_buffers_hook.py 4 additions, 4 deletionstests/test_hook/test_sync_buffers_hook.py
- tests/test_runner/test_runner.py 10 additions, 11 deletionstests/test_runner/test_runner.py
Loading
Please register or sign in to comment