Add --verbose option to infer_tool (#6770)
Detailed infer results printing overload output and freeze sea_runtool. Hiding debug messages by default.
This commit is contained in:
@@ -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!")
|
||||
|
||||
Reference in New Issue
Block a user