 czzhangheng
|
a9313390ac
|
适配GraphWaveNet
|
2025-12-03 12:05:02 +08:00 |
|
czzhangheng
|
c22d946393
|
mask_mae
|
2025-11-27 08:36:07 +08:00 |
|
czzhangheng
|
4d87087147
|
测试时汇总样本 使用detach到gpu 避免显存爆炸
|
2025-11-20 21:21:39 +08:00 |
|
czzhangheng
|
96f2ea1239
|
修复反归一化错误
|
2025-11-20 20:50:35 +08:00 |
|
czzhangheng
|
fe3fc186be
|
显示反归一化loss
|
2025-11-18 10:46:08 +08:00 |
|
czzhangheng
|
8b7e13df30
|
优化trainer和run
|
2025-11-18 10:01:01 +08:00 |
|
czzhangheng
|
2ba061e57a
|
修复confuse_layer硬编码bug
|
2025-11-11 17:26:05 +08:00 |
|
czzhangheng
|
9b3bb44552
|
修复配置, trainer
|
2025-11-09 22:30:23 +08:00 |
|
czzhangheng
|
241274e60f
|
删除冗余代码。重构代码
|
2025-11-09 20:40:19 +08:00 |
|
czzhangheng
|
cf47acf2bd
|
修复test bug
|
2025-11-09 16:38:04 +08:00 |
|
czzhangheng
|
b02c9c91d7
|
删除多余空行
|
2025-11-08 21:26:56 +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
|
bacdce7c72
|
trainer添加效率统计
|
2025-08-17 14:03:26 +08:00 |
|
czzhangheng
|
eb8684bf91
|
收到垃圾堆
|
2025-04-22 14:54:39 +08:00 |
|
HengZhang
|
26758e761b
|
init
|
2025-03-02 23:41:12 +08:00 |