DeepSpeed
Change zero_grad() argument to match pytorch
#2741
Merged
Overview
Commits
2
Changes
View On GitHub
Commits
  • Change argument set_grads_to_None to set_to_none to match pytorch convention. Also change default to False to match PyTorch
    loadams committed 3 years ago
  • PR feedback, changing internal usage since no-arg behavior is different now
    loadams committed 3 years ago
FAQ Terms Privacy Refunds Impressum
Loading