TrafficWheel/model
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 简化NLT, DDGCRN, GWN代码 2025-03-27 21:28:39 +08:00
DSANET init 2025-03-02 23:41:12 +08:00
GWN 简化NLT, DDGCRN, GWN代码 2025-03-27 21:28:39 +08:00
NLT 简化NLT, DDGCRN, GWN代码 2025-03-27 21:28:39 +08:00
PDG2SEQ add PDF2SeQ 2025-03-10 19:02:42 +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 mv dir name 2025-03-03 10:29:42 +08:00
STMLP 新增了模型蒸馏STMLP 2025-04-07 17:05:59 +08:00
STSGCN fix bugs 2025-03-03 16:31:04 +08:00
TCN init 2025-03-02 23:41:12 +08:00
TWDGCN init 2025-03-02 23:41:12 +08:00
model_selector.py 新增了模型蒸馏STMLP 2025-04-07 17:05:59 +08:00