Update hubconf.py attempt_load import (#4428)

pull/4436/head
Omid Sadeghnezhad 2021-08-16 13:28:02 +04:30 committed by GitHub
parent 19d03a955c
commit dbc06ce292
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions

View File

@ -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