DeepSpeed
Change zero_grad() argument to match pytorch
#2741
Merged

Change zero_grad() argument to match pytorch #2741

jeffra merged 2 commits into master from loadams/args-to-match-torch
loadams
loadams Change argument set_grads_to_None to set_to_none to match pytorch con…
2264fc11
loadams loadams marked this pull request as ready for review 2 years ago
loadams loadams requested a review from jeffra jeffra 2 years ago
loadams loadams requested a review from tjruwase tjruwase 2 years ago
loadams loadams requested a review from samyam samyam 2 years ago
loadams loadams requested a review from mrwyattii mrwyattii 2 years ago
tjruwase
tjruwase approved these changes on 2023-01-24
jeffra
loadams PR feedback, changing internal usage since no-arg behavior is differe…
07a4b567
jeffra
jeffra approved these changes on 2023-01-24
jeffra jeffra merged 34a11688 into master 2 years ago
jeffra jeffra deleted the loadams/args-to-match-torch branch 2 years ago

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone