Go
pytorch
25fa964d
- [shard] add clone/detach and set requires_grad for ShardedTensor
Commit
View On
GitHub
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
2 years ago
[shard] add clone/detach and set requires_grad for ShardedTensor This PR adding clone/detach and set requires_grad to ShardedTensor Pull Request resolved: https://github.com/pytorch/pytorch/pull/77367 Approved by: https://github.com/pritamdamania87
Author
wanchaol
Committer
pytorchmergebot
Parents
25c6ebd1
Loading