TrafficWheel/lib
czzhangheng bc9a2667c2 新增了模型蒸馏STMLP
现在Trainer每次epoch完后都会保存模型checkpoint
其中STMLP会自动教师模型到pre-train
根据教师模型的存在情况启动/预训练or蒸馏模式
2025-04-07 17:05:59 +08:00
..
Download_data.py 新增了模型蒸馏STMLP 2025-04-07 17:05:59 +08:00
Trainer_old.py init 2025-03-02 23:41:12 +08:00
initializer.py 添加固定种子 2025-04-04 14:37:05 +08:00
logger.py init 2025-03-02 23:41:12 +08:00
loss_function.py init 2025-03-02 23:41:12 +08:00
normalization.py fix bugs 2025-03-03 16:31:04 +08:00