diff --git a/python/text_utils/api/trainer.py b/python/text_utils/api/trainer.py index 82b9dd0..fbf5311 100644 --- 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,