Skip to content

添加:EDG++选择蒸馏 - #4

Open
katewangsiyu wants to merge 1 commit into
HongxinXiang:EDGppfrom
katewangsiyu:edgpp-for-pr
Open

添加:EDG++选择蒸馏#4
katewangsiyu wants to merge 1 commit into
HongxinXiang:EDGppfrom
katewangsiyu:edgpp-for-pr

Conversation

@katewangsiyu

Copy link
Copy Markdown

新增finetune_QM9_distillation.py和finetune_MD17_distillation.py蒸馏训练脚本
新增config_distillation.py配置文件,支持自适应阈值参
更新distillation_utils.py,实现选择性蒸馏机制

- 新增 finetune_QM9_distillation.py 和 finetune_MD17_distillation.py 蒸馏训练脚本
- 新增 config_distillation.py 配置文件,支持自适应阈值参数
- 更新 distillation_utils.py,实现选择性蒸馏机制
- 添加预训练模型目录结构(pretrained_IEMv2_models)
- 添加预训练代码(EDG/pretrain/)用于 ED-aware Teacher 训练
@katewangsiyu
katewangsiyu changed the base branch from master to EDGpp April 12, 2026 15:13
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant