megvii-research / mdistiller

The official implementation of [CVPR2022] Decoupled Knowledge Distillation https://arxiv.org/abs/2203.08679 and [ICCV2023] DOT: A Distillation-Oriented Trainer https://openaccess.thecvf.com/content/ICCV2023/papers/Zhao_DOT_A_Distillation-Oriented_Trainer_ICCV_2023_paper.pdf
808 stars 123 forks source link

DKD代码问题 #47

Closed xiaoxusanheyi closed 1 year ago

xiaoxusanheyi commented 1 year ago

你好作者:我想问一下DKD部分的代码具体在哪里那个位置,可否给指出来。 可以详细一点多少行开始到多少行结束。 祝好!

ufestkc commented 1 year ago

https://github.com/megvii-research/mdistiller/tree/master/mdistiller/distillers

不知道你是不是找的这个

xiaoxusanheyi commented 1 year ago

该目录下的DKD.py文件里的代码就是解耦知识蒸馏(DKD)的代码是吧!

ufestkc commented 1 year ago

对,主要的几个函数,缝起来就行了