glenn-jocher commited on
Commit
8ace1b1
·
unverified ·
1 Parent(s): 9f98201

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

Browse files

* YOLOv5 PyTorch Hub models >> check_requirements()

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

* Update hubconf.py

Files changed (1) hide show
  1. hubconf.py +1 -1
hubconf.py CHANGED
@@ -15,7 +15,7 @@ from utils.google_utils import attempt_download
15
  from utils.torch_utils import select_device
16
 
17
  dependencies = ['torch', 'yaml']
18
- check_requirements(exclude=('pycocotools', 'thop'))
19
  set_logging()
20
 
21
 
 
15
  from utils.torch_utils import select_device
16
 
17
  dependencies = ['torch', 'yaml']
18
+ check_requirements(Path(__file__).parent / 'requirements.txt', exclude=('pycocotools', 'thop'))
19
  set_logging()
20
 
21