Omid Sadeghnezhad
commited on
Commit
•
dbc06ce
1
Parent(s):
19d03a9
Update hubconf.py attempt_load import (#4428)
Browse files- hubconf.py +2 -1
hubconf.py
CHANGED
@@ -27,7 +27,8 @@ def _create(name, pretrained=True, channels=3, classes=80, autoshape=True, verbo
|
|
27 |
"""
|
28 |
from pathlib import Path
|
29 |
|
30 |
-
from models.yolo import Model
|
|
|
31 |
from utils.general import check_requirements, set_logging
|
32 |
from utils.downloads import attempt_download
|
33 |
from utils.torch_utils import select_device
|
|
|
27 |
"""
|
28 |
from pathlib import Path
|
29 |
|
30 |
+
from models.yolo import Model
|
31 |
+
from models.experimental import attempt_load
|
32 |
from utils.general import check_requirements, set_logging
|
33 |
from utils.downloads import attempt_download
|
34 |
from utils.torch_utils import select_device
|