Browse Source

Update hubconf.py attempt_load import (#4428)

modifyDataloader
Omid Sadeghnezhad GitHub 3 years ago
parent
commit
dbc06ce292
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
  1. +2
    -1
      hubconf.py

+ 2
- 1
hubconf.py 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

Loading…
Cancel
Save