Update check_requirements() exclude list (#2974)

2021-04-29 21:16:23 +02:00
1 parent dbce1bc54c
commit a833ee2a46
@@ -497,7 +497,7 @@ if __name__ == '__main__':
    set_logging(opt.global_rank)
    if opt.global_rank in [-1, 0]:
        check_git_status()
-        check_requirements()
+        check_requirements(exclude=('pycocotools', 'thop'))

    # Resume
    wandb_run = check_wandb_resume(opt)