remove debug comment

apax-hub · Apr 8, 2024 · c119992 · c119992
1 parent f84735a
commit c119992
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/apax/train/trainer.py b/apax/train/trainer.py
@@ -95,9 +95,6 @@ def fit(
             callbacks.on_train_batch_begin(batch=batch_idx)
 
             batch = next(batch_train_ds)
-
-            # print(jax.tree_map(lambda x: x.devices(), batch))
-
             (
                 (state, train_batch_metrics),
                 batch_loss,