pytorch
270c518b - [checkpoint] Implement interop between Tensor and Sharded Tensor (#78120)

Commit
2 years ago
[checkpoint] Implement interop between Tensor and Sharded Tensor (#78120) This allows loading a Tensor from a checkpoint with a SharedTensor in the same FQN. Pull Request resolved: https://github.com/pytorch/pytorch/pull/78120 Approved by: https://github.com/pritamdamania87
Author
Rodrigo Kumpera
Committer
Parents
Loading