Skip to content

Instantly share code, notes, and snippets.

@willprice

willprice/hub_demo.py

Created June 25, 2019 08:07
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save willprice/bb09973534bd5b09873efb58053d872e to your computer and use it in GitHub Desktop.
Save willprice/bb09973534bd5b09873efb58053d872e to your computer and use it in GitHub Desktop.
Download ZIP
Pytorch hub demo using pretorched-x
Raw
hub_demo.py
import torch
repo = "willprice/pretorched-x:pytorch-hub-support"
print(torch.hub.list(repo, force_reload=True))
model = torch.hub.load(repo, 'resneti3d50')
print(model)
print(dir(model))
print(help(model))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment