diff --git a/crates/ml/src/trainers/dqn/trainer/training_loop.rs b/crates/ml/src/trainers/dqn/trainer/training_loop.rs index dcdf2525f..37ebe0cae 100644 --- a/crates/ml/src/trainers/dqn/trainer/training_loop.rs +++ b/crates/ml/src/trainers/dqn/trainer/training_loop.rs @@ -1175,6 +1175,7 @@ impl DQNTrainer { } for _step in 0..num_training_steps { + if _step < 3 { eprintln!("H100_LOOP: step {_step}/{num_training_steps} — sampling PER"); } let sample_start = std::time::Instant::now(); let (batch, vaccine_batch) = { let agent = self.agent.read().await;