 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
|
3b4acd4951
|
合并
|
2025-12-14 17:50:08 +08:00 |
|
czzhangheng
|
5827554c73
|
改进模型注册，动态注册
|
2025-12-14 17:48:37 +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 |