mobile-vision/d2go 2/2

Summary: Pull Request resolved: https://github.com/facebookresearch/d2go/pull/645

Reviewed By: zsol

Differential Revision: D52536030

fbshipit-source-id: e6d0004c5bea81b5dab0ff69a1e9f6df4929b952

projects_oss/detr/main.py

@@ -513,7 +513,7 @@ def _distributed_worker(
     main_func(*args)
 
 
-if __name__ == "__main__":
+def invoke_main() -> None:
     parser = argparse.ArgumentParser(
         "DETR training and evaluation script", parents=[get_args_parser()]
     )
@@ -529,3 +529,7 @@ if __name__ == "__main__":
         dist_url=args.dist_url,
         args=(args,),
     )
+
+
+if __name__ == "__main__":
+    invoke_main()  # pragma: no cover