diff --git a/train.py b/train.py index 3f5b5ed1195b..250342acff18 100644 --- a/train.py +++ b/train.py @@ -451,7 +451,7 @@ def main(opt): if RANK in [-1, 0]: print(colorstr('train: ') + ', '.join(f'{k}={v}' for k, v in vars(opt).items())) check_git_status() - check_requirements(exclude=['thop']) + check_requirements(requirements=FILE.parent / 'requirements.txt', exclude=['thop']) # Resume if opt.resume and not check_wandb_resume(opt): # resume an interrupted run diff --git a/val.py b/val.py index 06b2501515b5..86439b1380dc 100644 --- a/val.py +++ b/val.py @@ -320,7 +320,7 @@ def parse_opt(): def main(opt): set_logging() print(colorstr('val: ') + ', '.join(f'{k}={v}' for k, v in vars(opt).items())) - check_requirements(exclude=('tensorboard', 'thop')) + check_requirements(requirements=FILE.parent / 'requirements.txt', exclude=('tensorboard', 'thop')) if opt.task in ('train', 'val', 'test'): # run normally run(**vars(opt))