pytorch
3d9cef8c - Clone tensor to write in ShardedTensor checkpoint (#79400)

Commit
3 years ago
Clone tensor to write in ShardedTensor checkpoint (#79400) The `torch.save` api will save the origin tensor of a view, which will results in saving a much larger checkpoint when parameters are fused, e.g. in torchrec. Relates to #79016 Pull Request resolved: https://github.com/pytorch/pytorch/pull/79400 Approved by: https://github.com/kumpera
Author
Committer
Parents
Loading