TrafficWheel/trainer
czzhangheng 3095b7435b refactor: 重构数据加载器和训练器代码,优化代码结构和可读性
重构数据加载器模块,使用字典映射替代switch-case结构
简化训练器逻辑,合并重复代码,提高可维护性
优化日志时间格式,缩短显示长度
调整训练配置,减少默认epoch数并启用GPU训练
统一数据加载方式,提取公共方法减少重复代码
2025-12-15 01:38:47 +08:00
..
cdeTrainer 测试所有配置是否正常运行 2025-12-01 22:29:52 +08:00
DCRNN_Trainer.py 更名ASTRA 2025-12-09 14:07:38 +08:00
E32Trainer.py 测试所有配置是否正常运行 2025-12-01 22:29:52 +08:00
EXP_trainer.py trainer修改 2025-12-01 21:36:37 +08:00
InformerTrainer.py 兼容InFormer 2025-12-11 23:16:25 +08:00
PDG2SEQ_Trainer.py 测试所有配置是否正常运行 2025-12-01 22:29:52 +08:00
STMLP_Trainer.py 测试所有配置是否正常运行 2025-12-01 22:29:52 +08:00
TSTrainer.py refactor: 重构数据加载器和训练器代码,优化代码结构和可读性 2025-12-15 01:38:47 +08:00
Trainer.py refactor: 重构数据加载器和训练器代码,优化代码结构和可读性 2025-12-15 01:38:47 +08:00
Trainer_bk.py 简化trainer,修复设备bug,新增批量运行脚本 2025-12-10 21:08:20 +08:00
trainer_selector.py refactor: 重构数据加载器和训练器代码,优化代码结构和可读性 2025-12-15 01:38:47 +08:00