zjc
/
yolov5
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

YOLOv5 PyTorch Hub models >> check_requirements() (#2588) 

* YOLOv5 PyTorch Hub models >> check_requirements()

Update YOLOv5 PyTorch Hub requirements.txt path to cache path.

* Update hubconf.py
This commit is contained in:
Glenn Jocher 2021-03-24 16:23:54 +01:00 committed by GitHub
parent 9f98201dd9
commit 8ace1b1b99
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
hubconf.py
View File

@ -15,7 +15,7 @@ from utils.google_utils import attempt_download
from utils.torch_utils import select_device from utils.torch_utils import select_device
dependencies = ['torch', 'yaml'] dependencies = ['torch', 'yaml']
check_requirements(exclude=('pycocotools', 'thop')) check_requirements(Path(__file__).parent / 'requirements.txt', exclude=('pycocotools', 'thop'))
set_logging() set_logging()