[Feature] Support EarlyStoppingHook (#739)
* [Feature] EarlyStoppingHook * delete redundant line * Assert stop_training and rename tests * Fix UT * rename `metric` to `monitor` * Fix UT * Fix UT * edit docstring on patience * Draft for new code * fix ut * add test case * add test case * fix ut * Apply suggestions from code review Co-authored-by:Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * Apply suggestions from code review Co-authored-by:
Mashiro <57566630+HAOCHENYE@users.noreply.github.com> * Append hook * Append hook * Apply suggestions * Update suggestions * Update mmengine/hooks/__init__.py * fix min_delta * Apply suggestions from code review * lint * Apply suggestions from code review Co-authored-by:
Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * delete save_last * infer rule more robust * refine unit test * Update mmengine/hooks/early_stopping_hook.py --------- Co-authored-by:
Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> Co-authored-by:
Mashiro <57566630+HAOCHENYE@users.noreply.github.com> Co-authored-by:
zhouzaida <zhouzaida@163.com> Co-authored-by:
HAOCHENYE <21724054@zju.edu.cn>
Showing
- docs/en/api/hooks.rst 1 addition, 0 deletionsdocs/en/api/hooks.rst
- docs/zh_cn/api/hooks.rst 1 addition, 0 deletionsdocs/zh_cn/api/hooks.rst
- mmengine/hooks/__init__.py 2 additions, 1 deletionmmengine/hooks/__init__.py
- mmengine/hooks/early_stopping_hook.py 159 additions, 0 deletionsmmengine/hooks/early_stopping_hook.py
- mmengine/runner/loops.py 8 additions, 2 deletionsmmengine/runner/loops.py
- tests/test_hooks/test_early_stopping_hook.py 255 additions, 0 deletionstests/test_hooks/test_early_stopping_hook.py
Loading
Please register or sign in to comment