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

* YOLOv5 PyTorch Hub models >> check_requirements() Update YOLOv5 PyTorch Hub requirements.txt path to cache path. * Update hubconf.py (cherry picked from commit 8ace1b1)
Lechtr · Mar 28, 2021 · 267fad6 · 267fad6
1 parent 6adb5ba
commit 267fad6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/hubconf.py b/hubconf.py
@@ -15,7 +15,7 @@
 from utils.torch_utils import select_device
 
 dependencies = ['torch', 'yaml']
-check_requirements(exclude=('pycocotools', 'thop'))
+check_requirements(Path(__file__).parent / 'requirements.txt', exclude=('pycocotools', 'thop'))
 set_logging()