From f238d1455823393b06dbaac864d7c25e788163ab Mon Sep 17 00:00:00 2001
From: Yizhuang Zhou <62599194+zhouyizhuang-megvii@users.noreply.github.com>
Date: Fri, 29 May 2020 14:37:51 +0800
Subject: [PATCH] fix(detection): fix hubconf.py import error (#26)

* fix(detection): fix hubconf.py import error

* fix(detection): fix hubconf.py import error
---
 hubconf.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/hubconf.py b/hubconf.py
index 775092f..3f0d37d 100644
--- a/hubconf.py
+++ b/hubconf.py
@@ -28,10 +28,10 @@ from official.nlp.bert.model import (
     wwm_cased_L_24_H_1024_A_16,
 )
 
-from official.vision.detection.retinanet_res50_1x_800size import (
-    retinanet_res50_1x_800size,
-    RetinaNet,
+from official.vision.detection.retinanet_res50_coco_1x_800size import (
+    retinanet_res50_coco_1x_800size,
 )
+from official.vision.detection.models import RetinaNet
 from official.vision.detection.tools.test import DetEvaluator
 
 from official.vision.segmentation.deeplabv3plus import (
-- 
GitLab