From 0dec8ffd66bbcbddcb65f40c7df8d645b03c4443 Mon Sep 17 00:00:00 2001
From: Lijun Yu <lijun@lj-y.com>
Date: Sun, 14 Jun 2020 14:47:40 -0400
Subject: [PATCH] path fix for torch hub

---
 hubconf.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/hubconf.py b/hubconf.py
index 729dac4..312e57b 100644
--- a/hubconf.py
+++ b/hubconf.py
@@ -6,6 +6,9 @@ Usage:
 """
 
 dependencies = ['torch', 'yaml']
+
+import os
+
 import torch
 
 from models.yolo import Model
@@ -24,7 +27,8 @@ def create(name, pretrained, channels, classes):
     Returns:
         pytorch model
     """
-    model = Model('models/%s.yaml' % name, channels, classes)
+    config = os.path.join(os.path.dirname(__file__), 'models', '%s.yaml' % name)
+    model = Model(config, channels, classes)
     if pretrained:
         ckpt = '%s.pt' % name  # checkpoint filename
         google_utils.attempt_download(ckpt)  # download if not found locally