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
KMint1819 · Mar 24, 2021 · 9d46a96 · 9d46a96
1 parent aa1c7bc
commit 9d46a96
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()