From a1b52d5dd01e9ee66a4795d43b9163f0d5380887 Mon Sep 17 00:00:00 2001 From: Omid Sadeghnezhad <58780720+OmidSa75@users.noreply.github.com> Date: Mon, 16 Aug 2021 13:28:02 +0430 Subject: [PATCH] Update hubconf.py attempt_load import (#4428) --- hubconf.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/hubconf.py b/hubconf.py index 36f3bd86bc11..799c83ec8400 100644 --- 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