Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

[3/n] loading meta to device #100495

Closed
wants to merge 1 commit into from
Closed

[3/n] loading meta to device #100495

wants to merge 1 commit into from

Commits on May 8, 2023

  1. [3/n] loading meta to device (pytorch#100495) 

    Summary:
Pull Request resolved: pytorch#100495

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: houseroad

Differential Revision: D45099145

fbshipit-source-id: a7e477e0632c55bfa4827a2183164f07ab1ee7da
    @qxy11 @facebook-github-bot
    qxy11 authored and facebook-github-bot committed May 8, 2023
    Configuration menu
    Copy the full SHA
    c968898 View commit details
    Browse the repository at this point in the history