From 6baedf9b4bd427cf08dee170d6c95637a124af41 Mon Sep 17 00:00:00 2001 From: jgrusewski Date: Wed, 8 Apr 2026 01:54:38 +0200 Subject: [PATCH] =?UTF-8?q?perf:=20vaccine=20every=2020th=20step=20(was=20?= =?UTF-8?q?10th)=20=E2=80=94=20same=20regularization,=20half=20the=20cost?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- crates/ml/src/trainers/dqn/fused_training.rs | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/crates/ml/src/trainers/dqn/fused_training.rs b/crates/ml/src/trainers/dqn/fused_training.rs index 252947389..dff4d169b 100644 --- a/crates/ml/src/trainers/dqn/fused_training.rs +++ b/crates/ml/src/trainers/dqn/fused_training.rs @@ -769,7 +769,7 @@ impl FusedTrainingCtx { } } // Gradient vaccine (1/10 steps). - if self.steps_since_varmap_sync % 10 == 0 { + if self.steps_since_varmap_sync % 20 == 0 { if let Some(ref vaccine_batch) = self.pending_vaccine_batch { if let Err(e) = self.trainer.apply_gradient_vaccine(vaccine_batch) { tracing::debug!("Gradient vaccine failed, restoring grad_buf: {e}");