TrafficWheel/utils
czzhangheng 3095b7435b refactor: 重构数据加载器和训练器代码,优化代码结构和可读性
重构数据加载器模块,使用字典映射替代switch-case结构
简化训练器逻辑,合并重复代码,提高可维护性
优化日志时间格式,缩短显示长度
调整训练配置,减少默认epoch数并启用GPU训练
统一数据加载方式,提取公共方法减少重复代码
2025-12-15 01:38:47 +08:00
..
Download_data.py AEPSA v0.1 2025-11-24 21:50:24 +08:00
__init__.py 收到垃圾堆 2025-04-22 14:54:39 +08:00
dataset.json 兼容NYCBike 2025-11-23 21:24:10 +08:00
get_adj.py 修复缺库问题 2025-11-10 11:13:39 +08:00
initializer.py 简化trainer,修复设备bug,新增批量运行脚本 2025-12-10 21:08:20 +08:00
logger.py refactor: 重构数据加载器和训练器代码,优化代码结构和可读性 2025-12-15 01:38:47 +08:00
loss_function.py mask_mae 2025-11-27 08:36:07 +08:00
normalization.py 删除冗余代码。重构代码 2025-11-09 20:40:19 +08:00
training_stats.py trainer修改 2025-12-01 21:36:37 +08:00