mmrazor/tests/test_engine/test_hooks
whcao 0919f69287
[feature] Support for specifying epochs to stop knowledge distillation (#455)
* support distillation early stop

* add norm arg to ChannelWiseDivergence

* add norm connector and support a list of connectors

* delete useless codes in cwd because of the usage of the norm connector

* fix fpn distill

* fix pytest

* rename stop distillation hook

* rename stop_epoch and add doc

* rename

* replace = with >=

* set _is_init private attribute of the teacher model to True after loading checkpoint
2023-03-01 17:10:22 +08:00
..
test_stop_distillation_hook.py [feature] Support for specifying epochs to stop knowledge distillation (#455) 2023-03-01 17:10:22 +08:00
test_visualization_hook.py [Feature]Feature map visualization (#293) 2022-10-26 13:26:20 +08:00