 czzhangheng
|
3095b7435b
|
refactor: 重构数据加载器和训练器代码，优化代码结构和可读性
重构数据加载器模块，使用字典映射替代switch-case结构
简化训练器逻辑，合并重复代码，提高可维护性
优化日志时间格式，缩短显示长度
调整训练配置，减少默认epoch数并启用GPU训练
统一数据加载方式，提取公共方法减少重复代码
|
2025-12-15 01:38:47 +08:00 |
|
czzhangheng
|
d8f4cc5825
|
简化trainer，修复设备bug，新增批量运行脚本
|
2025-12-10 21:08:20 +08:00 |
|
czzhangheng
|
5c2380ae21
|
适配HI
|
2025-12-09 17:49:18 +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
|
9c50c30918
|
AEPSA v0.1
|
2025-11-24 21:50:24 +08:00 |
|
czzhangheng
|
5f9ab76a27
|
兼容NYCBike
|
2025-11-23 21:24:10 +08:00 |
|
czzhangheng
|
cf1aa8022f
|
修复beijing taxi检测错误
|
2025-11-23 19:25:51 +08:00 |
|
czzhangheng
|
5cd81f4d4c
|
Merge branch 'REPST' of https://github.zhang-heng.com/czzhangheng/TrafficWheel into REPST
|
2025-11-23 19:10:51 +08:00 |
|
czzhangheng
|
95f81425b0
|
兼容consineAnnealineLR
|
2025-11-23 19:06:48 +08:00 |
|
czzhangheng
|
475a4788cd
|
兼容BJTaxi
|
2025-11-23 19:04:50 +08:00 |
|
czzhangheng
|
b7ea73bc92
|
更新数据集的检测与下载逻辑
|
2025-11-23 17:58:44 +08:00 |
|
czzhangheng
|
7055a6da64
|
更新正确的AirQuality数据集
|
2025-11-20 22:15:48 +08:00 |
|
czzhangheng
|
a46edc79a5
|
兼容BeijingAirQuality。重构data，需要更新pip requirement
|
2025-11-20 20:19:17 +08:00 |
|
czzhangheng
|
9911caa3d8
|
兼容solarEnergy
|
2025-11-20 11:28:58 +08:00 |
|
czzhangheng
|
76acc89499
|
兼容METR-LA
|
2025-11-20 10:48:05 +08:00 |
|
czzhangheng
|
2ba061e57a
|
修复confuse_layer硬编码bug
|
2025-11-11 17:26:05 +08:00 |
|
czzhangheng
|
d0af46ea5f
|
修复缺库问题
|
2025-11-10 11:13:39 +08:00 |
|
czzhangheng
|
241274e60f
|
删除冗余代码。重构代码
|
2025-11-09 20:40:19 +08:00 |
|
czzhangheng
|
221b67aa53
|
标准化PEP8格式
|
2025-11-08 21:26:03 +08:00 |
|
czzhangheng
|
eb8684bf91
|
收到垃圾堆
|
2025-04-22 14:54:39 +08:00 |