 czzhangheng
|
cc63e5078f
|
impl FPT
|
2025-12-19 10:19:17 +08:00 |
|
czzhangheng
|
21bc05e763
|
修复TSLoader Bug
|
2025-12-17 17:13:16 +08:00 |
|
czzhangheng
|
85257bc61c
|
refactor(trainer): 优化训练器代码结构并添加进度条显示
调整训练器代码结构，减少冗余代码，提高可读性
为训练过程添加tqdm进度条，实时显示loss信息
统一TRY环境变量的输出格式
简化日志记录和模型保存逻辑
|
2025-12-16 16:40:40 +08:00 |
|
czzhangheng
|
5e52f23c8d
|
fix(config): 修正配置文件命名不一致问题并更新训练参数
refactor(trainer): 重构训练器代码，优化反归一化处理和形状转换逻辑
style(trainer): 简化代码格式，提高可读性
chore: 更新训练脚本中的模型和数据集列表
|
2025-12-15 21:23:04 +08:00 |
|
czzhangheng
|
b6d4f5daf5
|
refactor(dataloader): 重构数据加载器代码，优化滑动窗口生成和归一化处理
重构PeMSD和EXP数据加载器，使用numpy的stride_tricks实现高效滑动窗口
统一数据预处理流程，简化代码结构并提高可维护性
优化归一化处理，支持多scaler通道独立处理
|
2025-12-15 20:54:20 +08:00 |
|
czzhangheng
|
56b09ea8ac
|
feat: 重构训练器和配置结构，优化模型训练流程
refactor(trainer): 重构Trainer类结构，拆分初始化方法
perf(trainer): 优化训练循环和评估逻辑
style(config): 统一配置文件命名和结构
fix(trainer): 修复形状检查逻辑和调试模式处理
docs: 更新README和注释说明
|
2025-12-15 19:55:57 +08:00 |
|
czzhangheng
|
3095b7435b
|
refactor: 重构数据加载器和训练器代码，优化代码结构和可读性
重构数据加载器模块，使用字典映射替代switch-case结构
简化训练器逻辑，合并重复代码，提高可维护性
优化日志时间格式，缩短显示长度
调整训练配置，减少默认epoch数并启用GPU训练
统一数据加载方式，提取公共方法减少重复代码
|
2025-12-15 01:38:47 +08:00 |
|
czzhangheng
|
97743dfd05
|
opt trainer
|
2025-12-15 00:38:43 +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
|
600420e8df
|
impl mtgnn
|
2025-12-10 23:31:17 +08:00 |
|
czzhangheng
|
4ccb029d7e
|
impl PatchTST
|
2025-12-10 21:53:46 +08:00 |
|
czzhangheng
|
d8f4cc5825
|
简化trainer，修复设备bug，新增批量运行脚本
|
2025-12-10 21:08:20 +08:00 |
|
czzhangheng
|
9c76975056
|
更名ASTRA
|
2025-12-09 14:07:38 +08:00 |
|
czzhangheng
|
a9313390ac
|
适配GraphWaveNet
|
2025-12-03 12:05:02 +08:00 |
|
czzhangheng
|
77a3210475
|
测试所有配置是否正常运行
|
2025-12-01 22:29:52 +08:00 |
|
czzhangheng
|
d4ee8e309e
|
trainer修改
|
2025-12-01 21:36:37 +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
|
af795043c8
|
Merge branch 'main' into dev
# Conflicts:
# run.py
# trainer/Trainer.py
|
2025-08-18 21:07:16 +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
|
4f7fb52707
|
更新EXP模型，添加周期性数据处理和时间特征，优化数据加载器和训练器，支持新的EXP32模型结构
|
2025-05-08 22:43:33 +08:00 |
|
czzhangheng
|
1be0b59344
|
新增EXP数据加载器、模型和训练器，支持周期性数据处理和动态图构建
|
2025-05-08 22:43:26 +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 |