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
ultralytics · Mar 24, 2021 · 8ace1b1 · 8ace1b1
1 parent 9f98201
commit 8ace1b1
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()