refactor exporter and eval command line tools

Summary: Tweak exporter and evaluator cli entry point func to support calling it as a module with args from custom launching code.

Reviewed By: sstsai-adl

Differential Revision: D35035813

fbshipit-source-id: c8b24099e94ccc58c184f8aac95b2a24a137e86a

tools/evaluator.py

@@ -7,6 +7,7 @@ torchscript, caffe2, etc.) using Detectron2Go system (dataloading, evaluation, e
 """
 
 import logging
+import sys
 
 import torch
 from d2go.distributed import launch
@@ -73,7 +74,7 @@ def run_with_cmdline_args(args):
     )
 
 
-if __name__ == "__main__":
+def cli(args):
     parser = basic_argument_parser()
     parser.add_argument(
         "--predictor-path",
@@ -100,3 +101,7 @@ if __name__ == "__main__":
         help="Control the --caffe2_logging_print_net_summary in GlobalInit",
     )
     run_with_cmdline_args(parser.parse_args())
+
+
+if __name__ == "__main__":
+    cli(sys.argv[1:])

tools/exporter.py

@@ -8,6 +8,7 @@ deployable format (such as torchscript, caffe2, ...)
 
 import copy
 import logging
+import sys
 import typing
 
 import mobile_cv.lut.lib.pt.flops_utils as flops_utils
@@ -125,9 +126,9 @@ def get_parser():
     return parser
 
 
-def cli():
-    run_with_cmdline_args(get_parser().parse_args())
+def cli(args):
+    run_with_cmdline_args(get_parser().parse_args(args))
 
 
 if __name__ == "__main__":
-    cli()
+    cli(sys.argv[1:])