"docker/vscode:/vscode.git/clone" did not exist on "1543cee7c8c95ef47f832b1f37625ba2923c4994"
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:
"""
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):
"""
......
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