diff --git a/hubconf.py b/hubconf.py index 4b4ae04cf332..710882cf158f 100644 --- 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()