 czzhangheng
|
a3fb2d5ec6
|
更新AEPSA框架，待修改文件
|
2025-11-11 21:29:52 +08:00 |
|
czzhangheng
|
2ba061e57a
|
修复confuse_layer硬编码bug
|
2025-11-11 17:26:05 +08:00 |
|
czzhangheng
|
f3480fccdc
|
STID配置更新
|
2025-11-10 09:34:52 +08:00 |
|
czzhangheng
|
9b3bb44552
|
修复配置, trainer
|
2025-11-09 22:30:23 +08:00 |
|
czzhangheng
|
2685d049d7
|
兼容pems-bay
|
2025-11-09 18:51:47 +08:00 |
|
czzhangheng
|
cf47acf2bd
|
修复test bug
|
2025-11-09 16:38:04 +08:00 |
|
czzhangheng
|
a3e43fc6df
|
添加支持REPST In D8
|
2025-11-09 16:27:32 +08:00 |
|
czzhangheng
|
221b67aa53
|
标准化PEP8格式
|
2025-11-08 21:26:03 +08:00 |
|
czzhangheng
|
9f9abd9d1b
|
预备升级为2.0
|
2025-11-08 21:05:52 +08:00 |
|
czzhangheng
|
29fd709c8c
|
新增多个PEMS数据集配置文件，包含PEMSD3、PEMSD4、PEMSD7、PEMSD8及STAWnet、STGNRDE、ST_SSL模型的相关配置，优化模型训练参数设置。
|
2025-08-19 00:07:37 +08:00 |
|
czzhangheng
|
b820b867fb
|
新模型
|
2025-08-19 00:07:13 +08:00 |
|
czzhangheng
|
bef30b9c2f
|
解决合并冲突，整合dev和main分支的更改
|
2025-05-14 13:13:11 +08:00 |
|
czzhangheng
|
4f7fb52707
|
更新EXP模型，添加周期性数据处理和时间特征，优化数据加载器和训练器，支持新的EXP32模型结构
|
2025-05-08 22:43:33 +08:00 |
|
czzhangheng
|
e826240a5e
|
Merge remote-tracking branch 'origin/main'
# Conflicts:
# model/model_selector.py
|
2025-04-23 23:24:05 +08:00 |
|
czzhangheng
|
97eb39073a
|
添加STIDGCN
|
2025-04-23 23:22:50 +08:00 |
|
czzhangheng
|
eb8684bf91
|
收到垃圾堆
|
2025-04-22 14:54:39 +08:00 |
|
czzhangheng
|
e851eb21d6
|
e21-e26无改进
|
2025-04-21 20:31:09 +08:00 |
|
czzhangheng
|
0b006087ea
|
e19添加时间嵌入
|
2025-04-18 14:59:47 +08:00 |
|
czzhangheng
|
bd94d3fdd3
|
exp12 残差
|
2025-04-18 10:20:55 +08:00 |
|
czzhangheng
|
86fabd4ca7
|
增加 exp9 混合专家
exp8 动态图manba
|
2025-04-17 18:41:57 +08:00 |
|
czzhangheng
|
c9a5a54d90
|
dev分支的实验
|
2025-04-17 15:57:39 +08:00 |
|
czzhangheng
|
bc9a2667c2
|
新增了模型蒸馏STMLP
现在Trainer每次epoch完后都会保存模型checkpoint
其中STMLP会自动教师模型到pre-train
根据教师模型的存在情况启动/预训练or蒸馏模式
|
2025-04-07 17:05:59 +08:00 |
|
czzhangheng
|
dcd4f99f9c
|
fix PDG2SEQ config
|
2025-03-10 19:05:26 +08:00 |
|
czzhangheng
|
c07bf05324
|
add PDF2SeQ
|
2025-03-10 19:02:42 +08:00 |
|
czzhangheng
|
0a9ac1a025
|
fix bugs
|
2025-03-03 16:31:04 +08:00 |
|
czzhangheng
|
f35bfbf75d
|
fix DSANET
|
2025-03-03 15:41:14 +08:00 |
|
HengZhang
|
44517219b7
|
add dl function
|
2025-03-03 11:34:36 +08:00 |
|
HengZhang
|
26758e761b
|
init
|
2025-03-02 23:41:12 +08:00 |