Explicit requirements.txt location (#4225)
This commit is contained in:
1 parent
7820614c40
commit
c2c958c350
2 files changed
+2
-2
No files matched your search
@@ -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
|
||||
|
||||
@@ -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))
|
||||
|
||||
Reference in new issue
Block a user