2 Commits

Author	SHA1	Message	Date
czzhangheng	8958cd7d95	更新.gitignore以忽略.temp_repo文件，更新README.md以支持新模型STAWnet，修改DCRNN和STGNRDE配置文件，添加STAWnet和STGNRDE模型及其训练器，增强模型选择器和数据加载器以支持新模型，更新损失函数以适应新需求，添加训练统计功能以跟踪训练过程中的性能指标。	2025-08-18 23:58:19 +08:00
czzhangheng	bc9a2667c2	新增了模型蒸馏STMLP ... 现在Trainer每次epoch完后都会保存模型checkpoint 其中STMLP会自动教师模型到pre-train 根据教师模型的存在情况启动/预训练or蒸馏模式	2025-04-07 17:05:59 +08:00