Torch Hub Support

README.md

@@ -37,6 +37,15 @@ Conference
 ## Description   
 What it does   
 
+## Use From Torch Hub
+```python
+import torch
+model = torch.hub.load(
+    "pytorchlightning/deep-learning-project-template",
+    "lit_classifier",
+)
+```
+
 ## How to run   
 First, install dependencies   
 ```bash

hubconf.py

@@ -0,0 +1,8 @@
+from project.lit_mnist import LitClassifier
+
+
+# any function defined here will be able to load a model like:
+# torch.hub.load("username/repo", "lit_classifier", *args, **kwargs)
+# can put logic here for loading pretrained weights
+def lit_classifier(*args, **kwargs):
+    return LitClassifier(*args, **kwargs)