Commit Graph

6 Commits

Author SHA1 Message Date
czzhangheng bc9a2667c2 新增了模型蒸馏STMLP
现在Trainer每次epoch完后都会保存模型checkpoint
其中STMLP会自动教师模型到pre-train
根据教师模型的存在情况启动/预训练or蒸馏模式
2025-04-07 17:05:59 +08:00
czzhangheng e8fc67b867 简化NLT, DDGCRN, GWN代码 2025-03-27 21:28:39 +08:00
HengZhang 44517219b7 add dl function 2025-03-03 11:34:36 +08:00
HengZhang be5e810c54 mv dir name 2025-03-03 10:29:42 +08:00
HengZhang 26758e761b init 2025-03-02 23:41:12 +08:00
czzhangheng 3f77b2759b Initial commit 2025-03-02 23:39:13 +08:00