forgot to remvoe print

ad-freiburg · Jun 26, 2024 · d7d7178 · d7d7178
1 parent 2a25309
commit d7d7178
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/python/text_utils/api/trainer.py b/python/text_utils/api/trainer.py
@@ -1094,8 +1094,6 @@ def _train_one_epoch(self):
                     (end_preparation - start_preparation) * 1000
                 )
 
-                print(len(batch), rank_batch_size)
-
                 with torch.autocast(
                     "cuda",
                     dtype=self.mixed_precision,