From e0a62ccbb5a606838c914d9a225c0a8ac306daa0 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sat, 18 Feb 2023 15:47:31 -0600 Subject: [PATCH] better ONNX optimization logging --- api/onnx_web/params.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/api/onnx_web/params.py b/api/onnx_web/params.py index 59e911ff..623dd02b 100644 --- a/api/onnx_web/params.py +++ b/api/onnx_web/params.py @@ -108,13 +108,17 @@ class DeviceParams: sess.enable_mem_reuse = False if "onnx-optimization-disable" in self.optimizations: + logger.debug("disabling all ONNX graph optimizations") sess.graph_optimization_level = GraphOptimizationLevel.ORT_DISABLE_ALL elif "onnx-optimization-basic" in self.optimizations: + logger.debug("enabling basic ONNX graph optimizations") sess.graph_optimization_level = GraphOptimizationLevel.ORT_ENABLE_BASIC elif "onnx-optimization-all" in self.optimizations: + logger.debug("enabling all ONNX graph optimizations") sess.graph_optimization_level = GraphOptimizationLevel.ORT_ENABLE_ALL if "onnx-deterministic-compute" in self.optimizations: + logger.debug("enabling ONNX deterministic compute") sess.use_deterministic_compute = True return sess