diff options
| author | Yuren Hao <yurenh2@illinois.edu> | 2026-07-09 03:43:54 -0500 |
|---|---|---|
| committer | Yuren Hao <yurenh2@illinois.edu> | 2026-07-09 03:43:54 -0500 |
| commit | a015da4a613f71f23dcfe2168c7d783faa0a6c55 (patch) | |
| tree | 82df8436e88b071cc7e1ed513d023bf04e6469d5 /ep_run/lt_ep_train.py | |
| parent | 77bb710f6c7b7d2a820a45daa99e43561553b22d (diff) | |
lt_ep_train: --gov_keepjr now keeps the jr floor through reg_delay too (fastfull arm: jr always-on, resreg post-warmup)
Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
Claude-Session: https://claude.ai/code/session_014FAPDWQ49M5Ye3NpTndTpn
Diffstat (limited to 'ep_run/lt_ep_train.py')
| -rw-r--r-- | ep_run/lt_ep_train.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ep_run/lt_ep_train.py b/ep_run/lt_ep_train.py index 58c18e9..9fb2c4a 100644 --- a/ep_run/lt_ep_train.py +++ b/ep_run/lt_ep_train.py @@ -691,7 +691,7 @@ def main(): if cfg.mode == 'ep': sw = hw_swap() if hw_on else None dly = (step < cfg.reg_delay) or (cfg.governor and not gov['engaged']) - _jr = jr if (cfg.gov_keepjr and not (step < cfg.reg_delay)) else (0.0 if dly else jr) + _jr = jr if cfg.gov_keepjr else (0.0 if dly else jr) # keepjr: jr floor survives BOTH gates (reg_delay & governor free phase) _rr = cfg.resreg * (cfg.gov_boost if step < gov['boost_until'] else 1.0) grads, res = ep_step(blk, idx, y, cfg.T1, cfg.T2, cfg.eps, cfg.beta, _jr, cfg.holo, cfg.hr, |