mobile-vision/experimental 2/5

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

Reviewed By: wat3rBro

Differential Revision: D52536555

fbshipit-source-id: e57dc5b2774771f0739118c5244014171732c151

tools/train_net.py

@@ -195,5 +195,9 @@ def build_cli_args(
     return args
 
 
-if __name__ == "__main__":
+def invoke_main() -> None:
     gather_mast_errors(cli())
+
+
+if __name__ == "__main__":
+    invoke_main()  # pragma: no cover