TrafficWheel/config
czzhangheng bc9a2667c2 新增了模型蒸馏STMLP
现在Trainer每次epoch完后都会保存模型checkpoint
其中STMLP会自动教师模型到pre-train
根据教师模型的存在情况启动/预训练or蒸馏模式
2025-04-07 17:05:59 +08:00
..
AGCRN init 2025-03-02 23:41:12 +08:00
ARIMA init 2025-03-02 23:41:12 +08:00
DCRNN init 2025-03-02 23:41:12 +08:00
DDGCRN init 2025-03-02 23:41:12 +08:00
DSANET fix DSANET 2025-03-03 15:41:14 +08:00
GWN init 2025-03-02 23:41:12 +08:00
NLT fix bugs 2025-03-03 16:31:04 +08:00
PDG2SEQ fix PDG2SEQ config 2025-03-10 19:05:26 +08:00
STFGNN init 2025-03-02 23:41:12 +08:00
STGCN init 2025-03-02 23:41:12 +08:00
STGNCDE init 2025-03-02 23:41:12 +08:00
STGODE init 2025-03-02 23:41:12 +08:00
STMLP 新增了模型蒸馏STMLP 2025-04-07 17:05:59 +08:00
STSGCN init 2025-03-02 23:41:12 +08:00
TCN init 2025-03-02 23:41:12 +08:00
TWDGCN init 2025-03-02 23:41:12 +08:00
args_parser.py add dl function 2025-03-03 11:34:36 +08:00