AstrAI/astrai/trainer
ViperEkura 70c0e5de90 refactor: merge validation into MetricCallback, simplify progress bar to optimizer steps
- Remove separate ValidationCallback, merge into MetricCallback
- Progress bar now tracks optimizer steps instead of micro-steps
- Remove unused log_interval config field and CLI flag
- Fix validation all_reduce: use SUM(loss, count) instead of AVG
- Simplify metric logging: always log every optimizer step
- Add grad_norm display to progress bar
2026-07-03 21:43:08 +08:00
..
__init__.py chore : 升级 torch 2.11.0+cu128,移除自定义 Muon,修复 gloo device_id 2026-06-27 16:10:37 +08:00
metric_util.py refactor : grad_norm 指标简化,clip_grad_norm 移至 executor 2026-06-30 14:59:43 +08:00
schedule.py fix: unify scheduler min_rate default to 0.01, clamp WSD warmup 2026-07-03 17:52:23 +08:00
strategy.py fix : 修复策略相关文件的类型注解与抽象方法体 2026-06-14 16:20:10 +08:00
train_callback.py refactor: merge validation into MetricCallback, simplify progress bar to optimizer steps 2026-07-03 21:43:08 +08:00
train_context.py refactor : replace iteration with consumed_samples 2026-06-30 18:42:42 +08:00
trainer.py refactor: merge validation into MetricCallback, simplify progress bar to optimizer steps 2026-07-03 21:43:08 +08:00