6 Commits

Author	SHA1	Message	Date
czzhangheng	77a3210475	测试所有配置是否正常运行	2025-12-01 22:29:52 +08:00
czzhangheng	d4ee8e309e	trainer修改	2025-12-01 21:36:37 +08:00
czzhangheng	241274e60f	删除冗余代码。重构代码	2025-11-09 20:40:19 +08:00
czzhangheng	221b67aa53	标准化PEP8格式	2025-11-08 21:26:03 +08:00
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