fix(gradient_logger): use layer.name→var mapping for Keras 3 compatibility 94e09b3 lemousehunter commited on 23 days ago
fix(gradient_logger): support LMIC-TSBN layer naming in _classify_layer cda2b12 lemousehunter commited on 23 days ago
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 23 days ago