Explicit requirements.txt location (#4225)

This commit is contained in:
Glenn Jocher authored and GitHub committed 2021-07-29 17:29:39 +02:00
1 parent 7820614c40
commit c2c958c350
2 files changed
+2 -2

No files matched your search

+1 -1
View File
@@ -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))