pytorch
ee1d375b
- [FSDP] Add fast path for `NO_SHARD` `clip_grad_norm_()` (#89137)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
2 years ago
[FSDP] Add fast path for `NO_SHARD` `clip_grad_norm_()` (#89137) Pull Request resolved: https://github.com/pytorch/pytorch/pull/89137 Approved by: https://github.com/rohan-varma
Author
awgu
Committer
pytorchmergebot
Parents
e70f446a
Loading