From a015da4a613f71f23dcfe2168c7d783faa0a6c55 Mon Sep 17 00:00:00 2001 From: Yuren Hao Date: Thu, 9 Jul 2026 03:43:54 -0500 Subject: 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 Claude-Session: https://claude.ai/code/session_014FAPDWQ49M5Ye3NpTndTpn --- ep_run/lt_ep_train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'ep_run/lt_ep_train.py') 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, -- cgit v1.2.3