TrafficWheel/config
czzhangheng 85257bc61c refactor(trainer): 优化训练器代码结构并添加进度条显示
调整训练器代码结构,减少冗余代码,提高可读性
为训练过程添加tqdm进度条,实时显示loss信息
统一TRY环境变量的输出格式
简化日志记录和模型保存逻辑
2025-12-16 16:40:40 +08:00
..
AGCRN 为model添加num_nodes 2025-12-03 08:40:00 +08:00
ARIMA 为model添加num_nodes 2025-12-03 08:40:00 +08:00
ASTRA feat: 重构训练器和配置结构,优化模型训练流程 2025-12-15 19:55:57 +08:00
ASTRA_v2 refactor(trainer): 优化训练器代码结构并添加进度条显示 2025-12-16 16:40:40 +08:00
ASTRA_v3 feat: 重构训练器和配置结构,优化模型训练流程 2025-12-15 19:55:57 +08:00
DCRNN 为model添加num_nodes 2025-12-03 08:40:00 +08:00
DDGCRN 为model添加num_nodes 2025-12-03 08:40:00 +08:00
DSANET 为model添加num_nodes 2025-12-03 08:40:00 +08:00
EXP 为model添加num_nodes 2025-12-03 08:40:00 +08:00
EXPB 为model添加num_nodes 2025-12-03 08:40:00 +08:00
GWN refactor(trainer): 优化训练器代码结构并添加进度条显示 2025-12-16 16:40:40 +08:00
HI refactor(config): 统一配置文件命名并调整模型参数 2025-12-15 21:33:28 +08:00
Informer fix(config): 修正配置文件命名不一致问题并更新训练参数 2025-12-15 21:23:04 +08:00
MTGNN refactor(trainer): 优化训练器代码结构并添加进度条显示 2025-12-16 16:40:40 +08:00
MegaCRN 为model添加num_nodes 2025-12-03 08:40:00 +08:00
NLT 为model添加num_nodes 2025-12-03 08:40:00 +08:00
PDG2SEQ 为model添加num_nodes 2025-12-03 08:40:00 +08:00
PatchTST refactor(config): 统一配置文件命名并调整模型参数 2025-12-15 21:33:28 +08:00
REPST refactor(trainer): 优化训练器代码结构并添加进度条显示 2025-12-16 16:40:40 +08:00
STAEFormer 更新配置,METR-LA -STAEFormer 2025-12-03 18:51:54 +08:00
STAWnet 为model添加num_nodes 2025-12-03 08:40:00 +08:00
STFGNN 为model添加num_nodes 2025-12-03 08:40:00 +08:00
STGCN 为model添加num_nodes 2025-12-03 08:40:00 +08:00
STGNCDE 为model添加num_nodes 2025-12-03 08:40:00 +08:00
STGNRDE 为model添加num_nodes 2025-12-03 08:40:00 +08:00
STGODE 为model添加num_nodes 2025-12-03 08:40:00 +08:00
STID refactor(config): 统一配置文件命名并调整模型参数 2025-12-15 21:33:28 +08:00
STIDGCN 为model添加num_nodes 2025-12-03 08:40:00 +08:00
STMLP 为model添加num_nodes 2025-12-03 08:40:00 +08:00
STSGCN 为model添加num_nodes 2025-12-03 08:40:00 +08:00
ST_SSL 为model添加num_nodes 2025-12-03 08:40:00 +08:00
TCN 为model添加num_nodes 2025-12-03 08:40:00 +08:00
TWDGCN 为model添加num_nodes 2025-12-03 08:40:00 +08:00
iTransformer fix(config): 修正配置文件命名不一致问题并更新训练参数 2025-12-15 21:23:04 +08:00
args_parser.py 简化trainer,修复设备bug,新增批量运行脚本 2025-12-10 21:08:20 +08:00