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

..

benchmark.py

fix: update benchmark to use checkpoint loading and CudaBackend

2026-07-30 22:54:45 +08:00

generate.py

refactor: remove H5 storage backend in favor of mmap bin

2026-07-29 12:50:27 +08:00

preprocess.py

refactor: remove H5 storage backend in favor of mmap bin

2026-07-29 12:50:27 +08:00

server.py

feat : add setup_logging with hierarchical astrai logger

2026-07-27 08:13:48 +08:00

train.py

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

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