From 122d7dc34fd0e397a08b8a584a632fc57d3fd5d0 Mon Sep 17 00:00:00 2001 From: Ibraheem Moosa Date: Thu, 8 Jul 2021 04:05:47 +0600 Subject: [PATCH] Remove logging of GPU count etc logging. (#12569) Successfully logging this requires Pytorch. For the purposes of this script we are not using Pytorch. --- examples/flax/language-modeling/run_t5_mlm_flax.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/examples/flax/language-modeling/run_t5_mlm_flax.py b/examples/flax/language-modeling/run_t5_mlm_flax.py index c8b6fa3da6..0cf6f3d632 100755 --- a/examples/flax/language-modeling/run_t5_mlm_flax.py +++ b/examples/flax/language-modeling/run_t5_mlm_flax.py @@ -430,10 +430,6 @@ if __name__ == "__main__": # Log on each process the small summary: logger = logging.getLogger(__name__) - logger.warning( - f"Process rank: {training_args.local_rank}, device: {training_args.device}, n_gpu: {training_args.n_gpu}" - + f"distributed training: {bool(training_args.local_rank != -1)}, 16-bits training: {training_args.fp16}" - ) # Set the verbosity to info of the Transformers logger (on main process only): logger.info(f"Training/evaluation parameters {training_args}")