 czzhangheng
|
04f65856f6
|
更新 .gitignore
|
2025-08-19 00:01:00 +08:00 |
|
czzhangheng
|
42b9cd3342
|
Merge remote-tracking branch 'origin/main'
|
2025-08-18 21:07:49 +08:00 |
|
czzhangheng
|
af795043c8
|
Merge branch 'main' into dev
# Conflicts:
# run.py
# trainer/Trainer.py
|
2025-08-18 21:07:16 +08:00 |
harry.zhang
|
8025a46baa
|
更新.gitignore以忽略Result.xlsx文件，修改run.py以优先使用macOS的MPS设备，优化设备设置逻辑
|
2025-08-18 14:32:20 +08:00 |
|
czzhangheng
|
bacdce7c72
|
trainer添加效率统计
|
2025-08-17 14:03:26 +08:00 |
|
czzhangheng
|
bef30b9c2f
|
解决合并冲突，整合dev和main分支的更改
|
2025-05-14 13:13:11 +08:00 |
|
czzhangheng
|
61565cd33a
|
为GraphWaveNet模型添加详细注释，增强代码可读性和可维护性
|
2025-05-14 13:09:20 +08:00 |
|
czzhangheng
|
4f7fb52707
|
更新EXP模型，添加周期性数据处理和时间特征，优化数据加载器和训练器，支持新的EXP32模型结构
|
2025-05-08 22:43:33 +08:00 |
|
czzhangheng
|
1be0b59344
|
新增EXP数据加载器、模型和训练器，支持周期性数据处理和动态图构建
|
2025-05-08 22:43:26 +08:00 |
|
czzhangheng
|
c15cf605be
|
添加md结果
|
2025-04-26 15:37:40 +08:00 |
|
czzhangheng
|
5f8c31af2e
|
添加STIDGCN
|
2025-04-23 23:24:43 +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
|
229b6320b9
|
添加固定种子
|
2025-04-04 14:37:05 +08:00 |
|
czzhangheng
|
e8fc67b867
|
简化NLT, DDGCRN, GWN代码
|
2025-03-27 21:28:39 +08:00 |
|
czzhangheng
|
1b76cc6ce2
|
优化GWN， ddgcrn的代码，使其更简洁易读
|
2025-03-27 20:08:17 +08:00 |
|
czzhangheng
|
d016dd5980
|
新增消耗分析模式，只需在原有的mode中调整为benchmark即可
|
2025-03-27 20:07:26 +08:00 |
|
czzhangheng
|
8c839642e1
|
新增消耗分析模式，只需在原有的mode中调整为benchmark即可
|
2025-03-27 20:02:16 +08:00 |
|
czzhangheng
|
5306d24408
|
新增消耗分析模式，只需在原有的mode中调整为benchmark即可
|
2025-03-26 12:38:57 +08:00 |
|
czzhangheng
|
dcd4f99f9c
|
fix PDG2SEQ config
|
2025-03-10 19:05:26 +08:00 |
|
czzhangheng
|
ae7e91d162
|
Merge remote-tracking branch 'origin/main'
|
2025-03-10 19:02:52 +08:00 |
|
czzhangheng
|
c07bf05324
|
add PDF2SeQ
|
2025-03-10 19:02:42 +08:00 |
|
HengZhang
|
7679507523
|
add url
|
2025-03-03 16:45:34 +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 |
|
czzhangheng
|
18aaa32c10
|
更新 transfer_guide.md
|
2025-03-03 15:25:51 +08:00 |
|
HengZhang
|
9181413d09
|
Update transfer_guide.md
|
2025-03-03 15:22:05 +08:00 |
|
HengZhang
|
01782a0c81
|
update Readme Add transfer_guide.md
|
2025-03-03 13:42:08 +08:00 |
|
HengZhang
|
5df1c810b1
|
edit README
|
2025-03-03 13:19:03 +08:00 |
|
HengZhang
|
d2b0ac5f9a
|
Merge branch 'main' of https://github.zhang-heng.com/czzhangheng/TrafficWheel
|
2025-03-03 13:01:21 +08:00 |
|
HengZhang
|
2e2b5608ae
|
add expResult
|
2025-03-03 12:59:34 +08:00 |
|
czzhangheng
|
d2a4a65dc6
|
develop pip
|
2025-03-03 11:54:23 +08:00 |
|
czzhangheng
|
085c4c2514
|
update pip
|
2025-03-03 11:52:09 +08:00 |
|
HengZhang
|
2bb113fed6
|
Update README.md
|
2025-03-03 11:38:32 +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 |