diff options
Diffstat (limited to 'ep_run')
| -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, |