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]:
|
if RANK in [-1, 0]:
|
||||||
print(colorstr('train: ') + ', '.join(f'{k}={v}' for k, v in vars(opt).items()))
|
print(colorstr('train: ') + ', '.join(f'{k}={v}' for k, v in vars(opt).items()))
|
||||||
check_git_status()
|
check_git_status()
|
||||||
check_requirements(exclude=['thop'])
|
check_requirements(requirements=FILE.parent / 'requirements.txt', exclude=['thop'])
|
||||||
|
|
||||||
# Resume
|
# Resume
|
||||||
if opt.resume and not check_wandb_resume(opt): # resume an interrupted run
|
if opt.resume and not check_wandb_resume(opt): # resume an interrupted run
|
||||||
|
|||||||
@@ -320,7 +320,7 @@ def parse_opt():
|
|||||||
def main(opt):
|
def main(opt):
|
||||||
set_logging()
|
set_logging()
|
||||||
print(colorstr('val: ') + ', '.join(f'{k}={v}' for k, v in vars(opt).items()))
|
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
|
if opt.task in ('train', 'val', 'test'): # run normally
|
||||||
run(**vars(opt))
|
run(**vars(opt))
|
||||||
|
|||||||
Reference in new issue
Block a user