Unverified Commit 9d8e8a87 authored by Sylvain Gugger's avatar Sylvain Gugger Committed by GitHub
Browse files

Avoid using no_sync on SageMaker DP (#11229)

parent 9fa29959
...@@ -805,7 +805,7 @@ class TrainingArguments: ...@@ -805,7 +805,7 @@ class TrainingArguments:
""" """
Whether or not to use no_sync for the gradients when doing gradient accumulation. Whether or not to use no_sync for the gradients when doing gradient accumulation.
""" """
return not (self.deepspeed or is_sagemaker_mp_enabled()) return not (self.deepspeed or is_sagemaker_dp_enabled() or is_sagemaker_mp_enabled())
def to_dict(self): def to_dict(self):
""" """
......
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