 czzhangheng
|
cc63e5078f
|
impl FPT
|
2025-12-19 10:19:17 +08:00 |
|
czzhangheng
|
3095b7435b
|
refactor: 重构数据加载器和训练器代码，优化代码结构和可读性
重构数据加载器模块，使用字典映射替代switch-case结构
简化训练器逻辑，合并重复代码，提高可维护性
优化日志时间格式，缩短显示长度
调整训练配置，减少默认epoch数并启用GPU训练
统一数据加载方式，提取公共方法减少重复代码
|
2025-12-15 01:38:47 +08:00 |
|
czzhangheng
|
9147803c2b
|
改进trainer，使用统一的loader而不是TSLoader，仅在Trainer上做了shape变换，确定尽可能少改动数据。
|
2025-12-14 17:47:38 +08:00 |
|
czzhangheng
|
19fd7622a3
|
兼容InFormer
|
2025-12-11 23:16:25 +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
|
bef30b9c2f
|
解决合并冲突，整合dev和main分支的更改
|
2025-05-14 13:13:11 +08:00 |
|
czzhangheng
|
4f7fb52707
|
更新EXP模型，添加周期性数据处理和时间特征，优化数据加载器和训练器，支持新的EXP32模型结构
|
2025-05-08 22:43:33 +08:00 |
|
czzhangheng
|
eb8684bf91
|
收到垃圾堆
|
2025-04-22 14:54:39 +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
|
c07bf05324
|
add PDF2SeQ
|
2025-03-10 19:02:42 +08:00 |
|
HengZhang
|
26758e761b
|
init
|
2025-03-02 23:41:12 +08:00 |