MCG-NJU/MixFormerV2

在训练student模型时,teacher模型权重加载问题

201722111112027 opened this issue · 1 comments

您好,在仔细研究了您的代码之后,本人发现,在训练student模型时,虽然涉及了参数checkpoint_teacher_path,但在代码中,teacher模型并未加载已训练好的权重。对此,本人不太理解。望您能对此进行解惑。

directory_teacher = self.settings.checkpoint_teacher_path

teacher 模型会在 trainer 中加载