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