Commit 744d72d7 authored by Tsahi Glik's avatar Tsahi Glik Committed by Facebook GitHub Bot
Browse files

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
parent b9dc151a
......@@ -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:])
......@@ -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:])
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment