Unverified Commit 99dbbdb9 authored by Riccardo Bassani's avatar Riccardo Bassani Committed by GitHub
Browse files

bugfixes training_args.py (#11922)

modified according to:
https://pytorch.org/xla/release/1.8.1/_modules/torch_xla/core/xla_model.html
parent 7e73601f
......@@ -861,7 +861,7 @@ class TrainingArguments:
The index of the local process used.
"""
if is_torch_tpu_available():
return xm.get_ordinal(local=True)
return xm.get_local_ordinal()
elif is_sagemaker_mp_enabled():
return smp.local_rank()
elif is_sagemaker_dp_enabled():
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment