2 Commits

Author	SHA1	Message	Date
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
whcao	1e8f886523	[Feature]Feature map visualization (#293) ... * WIP: vis * WIP: add visualization * WIP: add visualization hook * WIP: support razor visualizer * WIP * WIP: wrap draw_featmap * support feature map visualization * add a demo image for visualization * fix typos * change eps to 1e-6 * add pytest for visualization * fix vis hook * fix arguments' name * fix img path * support draw inference results * add visualization doc * fix figure url * move files Co-authored-by: weihan cao <HIT-cwh>	2022-10-26 13:26:20 +08:00