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
@@ -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
+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))