New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

在训练student模型时，teacher模型权重加载问题 #20

Open

201722111112027 opened this issue Feb 27, 2024 · 1 comment

201722111112027 commented Feb 27, 2024

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

Collaborator

songtianhui commented Feb 27, 2024

MixFormerV2/lib/train/trainers/base_trainer.py

Line 82 in 94ca2e6

directory_teacher = self.settings.checkpoint_teacher_path

teacher 模型会在 trainer 中加载

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment