Go
pytorch
c54f6302
- [7/n][FSDP] make use_dtensor=True work with offload_to_cpu=True for load_state_dict (#105378)
Commit
View On
GitHub
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
1 year ago
[7/n][FSDP] make use_dtensor=True work with offload_to_cpu=True for load_state_dict (#105378) Fixes #ISSUE_NUMBER Pull Request resolved: https://github.com/pytorch/pytorch/pull/105378 Approved by: https://github.com/fegin
Author
wz337
Committer
pytorchmergebot
Parents
5ce5372d
Loading