From 321625c3fb15580b9e76e296d077b6263dcd75d5 Mon Sep 17 00:00:00 2001
From: Teddy Koker <teddy.koker@gmail.com>
Date: Tue, 8 Dec 2020 14:49:55 -0500
Subject: [PATCH 1/4] add support for torchhub

---
 README.md  | 8 ++++++++
 hubconf.py | 6 ++++++
 2 files changed, 14 insertions(+)
 create mode 100644 hubconf.py

diff --git a/README.md b/README.md
index 63571ba..1c2588a 100644
--- a/README.md
+++ b/README.md
@@ -37,6 +37,14 @@ 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
diff --git a/hubconf.py b/hubconf.py
new file mode 100644
index 0000000..4a38849
--- /dev/null
+++ b/hubconf.py
@@ -0,0 +1,6 @@
+from project.lit_mnist import LitClassifier
+
+def lit_classifier(*args, **kwargs):
+    return LitClassifier(*args, **kwargs)
+
+

From db65d184649dd78854f896be9c062163b8ac12e1 Mon Sep 17 00:00:00 2001
From: Teddy Koker <teddy.koker@gmail.com>
Date: Tue, 8 Dec 2020 14:50:33 -0500
Subject: [PATCH 2/4] \n

---
 README.md | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 1c2588a..44688d1 100644
--- a/README.md
+++ b/README.md
@@ -42,7 +42,8 @@ What it does
 import torch
 model = torch.hub.load(
     "pytorchlightning/deep-learning-project-template",
-    "lit_classifier",)
+    "lit_classifier",
+)
 ```
 
 ## How to run   

From 03908d3df9e56e1a0a969c9d18a2f2c32f99ae2e Mon Sep 17 00:00:00 2001
From: Teddy Koker <teddy.koker@gmail.com>
Date: Tue, 8 Dec 2020 15:00:41 -0500
Subject: [PATCH 3/4] pep

---
 hubconf.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/hubconf.py b/hubconf.py
index 4a38849..b0db454 100644
--- a/hubconf.py
+++ b/hubconf.py
@@ -1,6 +1,5 @@
 from project.lit_mnist import LitClassifier
 
+
 def lit_classifier(*args, **kwargs):
     return LitClassifier(*args, **kwargs)
-
-

From 8928db3d6f65f9c12a9d195748fcb404787afee5 Mon Sep 17 00:00:00 2001
From: Teddy Koker <teddy.koker@gmail.com>
Date: Tue, 8 Dec 2020 15:03:36 -0500
Subject: [PATCH 4/4] comment

---
 hubconf.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/hubconf.py b/hubconf.py
index b0db454..aa8f256 100644
--- a/hubconf.py
+++ b/hubconf.py
@@ -1,5 +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)