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

Add device argument to PyTorch Hub models #3104

Merged
merged 3 commits into from
May 16, 2021
Merged

Add device argument to PyTorch Hub models #3104

merged 3 commits into from
May 16, 2021

Commits on May 10, 2021

  1. Allow to manual selection of device for torchhub models

    @cgerum
    cgerum committed May 10, 2021
    Configuration menu
    Copy the full SHA
    7bccd72 View commit details
    Browse the repository at this point in the history

Commits on May 16, 2021

  1. Merge branch 'master' into f/torchub_device_selection

    @glenn-jocher
    glenn-jocher committed May 16, 2021
    Configuration menu
    Copy the full SHA
    369a7a5 View commit details
    Browse the repository at this point in the history

  2. single line device 

    nested torch.device(torch.device(device)) ok
    @glenn-jocher
    glenn-jocher committed May 16, 2021
    Configuration menu
    Copy the full SHA
    767c502 View commit details
    Browse the repository at this point in the history