removed debug print statement

apax-hub · Jan 10, 2024 · e828e52 · e828e52
1 parent 4ecd6f9
commit e828e52
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/apax/data/input_pipeline.py b/apax/data/input_pipeline.py
@@ -252,7 +252,6 @@ def shuffle_and_batch(self) -> Iterator[jax.Array]:
         )
 
         if self.n_jit_steps > 1:
-            print("JIT BATCH")
             ds = ds.batch(batch_size=self.n_jit_steps)
 
         ds = prefetch_to_single_device(ds.as_numpy_iterator(), 2)