2 Commits

Author	SHA1	Message	Date
czzhangheng	29fd709c8c	新增多个PEMS数据集配置文件，包含PEMSD3、PEMSD4、PEMSD7、PEMSD8及STAWnet、STGNRDE、ST_SSL模型的相关配置，优化模型训练参数设置。	2025-08-19 00:07:37 +08:00
czzhangheng	bc9a2667c2	新增了模型蒸馏STMLP ... 现在Trainer每次epoch完后都会保存模型checkpoint 其中STMLP会自动教师模型到pre-train 根据教师模型的存在情况启动/预训练or蒸馏模式	2025-04-07 17:05:59 +08:00