Skip to content

GitLab

You need to sign in or sign up before continuing.
Unverified Commit 23ff6cb7 authored by Jeff Rasley's avatar Jeff Rasley Committed by GitHub
Browse files

update backward api doc (#903)

parent af2d8fc5
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
deepspeed/runtime/engine.py
View file @ 23ff6cb7
...@@ -955,7 +955,7 @@ class DeepSpeedEngine(Module): ...@@ -955,7 +955,7 @@ class DeepSpeedEngine(Module):
Arguments: Arguments:
loss: Torch tensor on which to execute backward propagation loss: Torch tensor on which to execute backward propagation
allreduce_gradients: If this is False, then gradient averaging will be skipped. Default is True. allreduce_gradients: is deprecated, ignored, and will soon be removed'
""" """
if not allreduce_gradients: if not allreduce_gradients:
......
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