pytorch
812cadf9 - [3/n] loading meta to device (#100495)

Commit
1 year ago
[3/n] loading meta to device (#100495) Summary: Make it possible to `torch.jit.load(model, device)` to a device when `model` contains weights that are on device `meta`. Just leave the `meta` weights on `meta`, and load the weights that can be loaded to the target device. Reviewed By: singlaiiit, RoshanPAN, sayitmemory Differential Revision: D45099145 Pull Request resolved: https://github.com/pytorch/pytorch/pull/100495 Approved by: https://github.com/houseroad
Author
Committer
Parents
Loading