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