diff --git a/tests/conditional_compilation/tools/infer_tool.py b/tests/conditional_compilation/tools/infer_tool.py index 9cc705d5c5f85d..9d733b6911cee9 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 @@ def cli_parser(): 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!")