ascad-training-pipeline / src /gradient_logger.py

Commit History

fix(gradient_logger): use layer.name→var mapping for Keras 3 compatibility
94e09b3

lemousehunter commited on

fix(gradient_logger): support LMIC-TSBN layer naming in _classify_layer
cda2b12

lemousehunter commited on

feat: add per-layer gradient norm logging to W&B — logs shared conv, per-task BN, head, gate norms every epoch
8c9d0fd

lemousehunter commited on