On-demand pycocotools pip install (#3547)

2021-06-09 11:22:21 +02:00
1 parent 1b5edb6f8e
commit ef0b5c9d29
@@ -495,7 +495,7 @@ if __name__ == '__main__':
    set_logging(opt.global_rank)
    if opt.global_rank in [-1, 0]:
        check_git_status()
-        check_requirements(exclude=('pycocotools', 'thop'))
+        check_requirements(exclude=['thop'])

    # Resume
    wandb_run = check_wandb_resume(opt)