Simplify check_requirements() usage (#4855)
* Simplify `check_requirements()` usage * remove assert, print()
This commit is contained in:
1 parent
4d1a2ac87e
commit
4c839eeb10
6 files changed
+9
-6
No files matched your search
+1
-1
@@ -34,7 +34,7 @@ def _create(name, pretrained=True, channels=3, classes=80, autoshape=True, verbo
|
||||
from utils.torch_utils import select_device
|
||||
|
||||
file = Path(__file__).resolve()
|
||||
check_requirements(requirements=file.parent / 'requirements.txt', exclude=('tensorboard', 'thop', 'opencv-python'))
|
||||
check_requirements(exclude=('tensorboard', 'thop', 'opencv-python'))
|
||||
set_logging(verbose=verbose)
|
||||
|
||||
save_dir = Path('') if str(name).endswith('.pt') else file.parent
|
||||
|
||||
Reference in new issue
Block a user