Update hubconf.py attempt_load import (ultralytics#4428)

CesarBazanAV · Sep 29, 2021 · a1b52d5 · a1b52d5
1 parent c1fd77d
commit a1b52d5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/hubconf.py b/hubconf.py
@@ -27,7 +27,8 @@ def _create(name, pretrained=True, channels=3, classes=80, autoshape=True, verbo
     """
     from pathlib import Path
 
-    from models.yolo import Model, attempt_load
+    from models.yolo import Model
+    from models.experimental import attempt_load
     from utils.general import check_requirements, set_logging
     from utils.downloads import attempt_download
     from utils.torch_utils import select_device