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