 czzhangheng
|
8958cd7d95
|
更新.gitignore以忽略.temp_repo文件，更新README.md以支持新模型STAWnet，修改DCRNN和STGNRDE配置文件，添加STAWnet和STGNRDE模型及其训练器，增强模型选择器和数据加载器以支持新模型，更新损失函数以适应新需求，添加训练统计功能以跟踪训练过程中的性能指标。
|
2025-08-18 23:58:19 +08:00 |
|
czzhangheng
|
bef30b9c2f
|
解决合并冲突，整合dev和main分支的更改
|
2025-05-14 13:13:11 +08:00 |
|
czzhangheng
|
eb8684bf91
|
收到垃圾堆
|
2025-04-22 14:54:39 +08:00 |
|
czzhangheng
|
bc9a2667c2
|
新增了模型蒸馏STMLP
现在Trainer每次epoch完后都会保存模型checkpoint
其中STMLP会自动教师模型到pre-train
根据教师模型的存在情况启动/预训练or蒸馏模式
|
2025-04-07 17:05:59 +08:00 |
|
czzhangheng
|
e8fc67b867
|
简化NLT, DDGCRN, GWN代码
|
2025-03-27 21:28:39 +08:00 |
|
HengZhang
|
44517219b7
|
add dl function
|
2025-03-03 11:34:36 +08:00 |
|
HengZhang
|
be5e810c54
|
mv dir name
|
2025-03-03 10:29:42 +08:00 |
|
HengZhang
|
26758e761b
|
init
|
2025-03-02 23:41:12 +08:00 |
|
czzhangheng
|
3f77b2759b
|
Initial commit
|
2025-03-02 23:39:13 +08:00 |