bugfixes training_args.py (#11922)
bugfixes training_args.py (#11922)

modified according to:
https://pytorch.org/xla/release/1.8.1/_modules/torch_xla/core/xla_model.html
huggingface/transformers,huggingface/transformers,huggingface/transformers,huggingface/transformers