ViperEkura d6bfb09863 feat: add grad_snr metric with EMA-based gradient SNR tracking ...

- add GradSNRTracker to metric_util.py computing SNR = E[g]^2 / Var(g) via per-parameter EMA moments
- add grad_snr_tracker field to TrainContext (instantiated by default)
- register grad_snr in MetricCallback, update tracker on each optimizer step before metrics are recorded
- add grad_snr to default --metrics in train.py CLI

2026-08-01 08:54:44 +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

feat: add grad_snr metric with EMA-based gradient SNR tracking

2026-08-01 08:54:44 +08:00

rollout.py

fix: correct online rollout lifecycle

2026-07-23 19:01:37 +08:00

schedule.py

fix: scheduler division-by-zero, loss_mask bool

2026-07-03 22:04:55 +08:00

strategy.py

fix: broadcast ref/old model state_dict for FSDP

2026-07-31 08:32:22 +08:00

train_callback.py

feat: add grad_snr metric with EMA-based gradient SNR tracking

2026-08-01 08:54:44 +08:00

train_context.py

feat: add grad_snr metric with EMA-based gradient SNR tracking

2026-08-01 08:54:44 +08:00

trainer.py

refactor: move signal_handler from parallel/ to top-level for broader reuse

2026-07-28 10:36:17 +08:00