SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Remove logging of GPU count etc logging. (#12569)

Browse Source 
Successfully logging this requires Pytorch. For the purposes of this script we are not using Pytorch.
This commit is contained in:
Ibraheem Moosa
2021-07-08 04:05:47 +06:00
committed by GitHub
parent d7e156bd1a
commit 122d7dc34f
1 changed files with 0 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
examples/flax/language-modeling/run_t5_mlm_flax.py
View File

@@ -430,10 +430,6 @@ if __name__ == "__main__":
# Log on each process the small summary: # Log on each process the small summary:
logger = logging.getLogger(__name__) 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): # Set the verbosity to info of the Transformers logger (on main process only):
logger.info(f"Training/evaluation parameters {training_args}") logger.info(f"Training/evaluation parameters {training_args}")