czzhangheng
/
TrafficWheel
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
53 Commits 3 Branches 0 Tags 187 MiB
Commit Graph

1 Commits

Author SHA1 Message Date
czzhangheng bc9a2667c2 新增了模型蒸馏STMLP 
现在Trainer每次epoch完后都会保存模型checkpoint
其中STMLP会自动教师模型到pre-train
根据教师模型的存在情况启动/预训练or蒸馏模式
 2025-04-07 17:05:59 +08:00