DeepSpeed
ba67bd9a - Added retain_graph as a kwarg to the main engine backward function (#1149)

Commit

3 years ago

Added retain_graph as a kwarg to the main engine backward function (#1149) Co-authored-by: Jeff Rasley <jerasley@microsoft.com>

References

Author

ncilfone

Parents