Lightning-AI / deep-learning-project-template

Pytorch Lightning code guideline for conferences
Apache License 2.0
1.23k stars 269 forks source link

Torch Hub Support #22

Open teddykoker opened 3 years ago

teddykoker commented 3 years ago

Add torch.hub support like so:

import torch
model = torch.hub.load(
    "pytorchlightning/deep-learning-project-template",
    "lit_classifier",
)
pep8speaks commented 3 years ago

Hello @teddykoker! Thanks for updating this PR. We checked the lines you've touched for PEP 8 issues, and found:

There are currently no PEP 8 issues detected in this Pull Request. Cheers! :beers:

Comment last updated at 2020-12-08 20:03:46 UTC
teddykoker commented 3 years ago

Build has been failing since previous commits to master, will fix in future pr