Omid Sadeghnezhad commited on
Commit
dbc06ce
1 Parent(s): 19d03a9

Update hubconf.py attempt_load import (#4428)

Browse files
Files changed (1) hide show
  1. 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, attempt_load
 
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