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