Open hopesala opened 2 weeks ago
我看cacti/utils/loss_builder.py,以及配置文件mobile_sci.py均未涉及到教师学生模型的蒸馏损失,请指点迷津,多谢~
Hi,知识蒸馏部分的代码还没有发布。另外,通过图中的公式5可以看出蒸馏的时候是将学生模型和老师模型的输出计算loss。然后和学生模型输出和GT计算出的loss相加得到训练的总loss(公式6)。
我看cacti/utils/loss_builder.py,以及配置文件mobile_sci.py均未涉及到教师学生模型的蒸馏损失,请指点迷津,多谢~