better ONNX optimization logging

ssube · Feb 18, 2023 · e0a62cc · e0a62cc
1 parent 881b290
commit e0a62cc
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/api/onnx_web/params.py b/api/onnx_web/params.py
@@ -108,13 +108,17 @@ def sess_options(self) -> SessionOptions:
             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