From 33dfcd62c2e9553e700917966800a4c3f3de6118 Mon Sep 17 00:00:00 2001 From: Ilya Sharikov Date: Mon, 26 Jul 2021 09:00:56 +0300 Subject: [PATCH] Add --verbose option to infer_tool (#6770) Detailed infer results printing overload output and freeze sea_runtool. Hiding debug messages by default. --- tests/conditional_compilation/tools/infer_tool.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/tests/conditional_compilation/tools/infer_tool.py b/tests/conditional_compilation/tools/infer_tool.py index 9cc705d5c5f..9d733b6911c 100644 --- a/tests/conditional_compilation/tools/infer_tool.py +++ b/tests/conditional_compilation/tools/infer_tool.py @@ -62,10 +62,14 @@ def cli_parser(): parser.add_argument('-d', dest='device', required=True, help='Target device to infer on') parser.add_argument('-r', dest='out_path', required=True, help='Dumps results to the output file') + parser.add_argument('-v', '--verbose', dest='verbose', action='store_true', + help='Increase output verbosity') args = parser.parse_args() ir_path = args.ir_path device = args.device out_path = args.out_path + if args.verbose: + log.getLogger().setLevel(log.DEBUG) return ir_path, device, out_path @@ -74,6 +78,6 @@ if __name__ == "__main__": results = infer(ir_path=ir_path, device=device) np.savez(out_path, **results) log.info("Path for inference results: {}".format(out_path)) - log.info("Inference results:") - log.info(results) - log.info("SUCCESS!") + log.debug("Inference results:") + log.debug(results) + log.debug("SUCCESS!")