apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

better ONNX optimization logging

This commit is contained in:
Sean Sube 2023-02-18 15:47:31 -06:00
parent 881b290116
commit e0a62ccbb5
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/onnx_web/params.py
View File

@ -108,13 +108,17 @@ class DeviceParams:
sess.enable_mem_reuse = False sess.enable_mem_reuse = False
if "onnx-optimization-disable" in self.optimizations: if "onnx-optimization-disable" in self.optimizations:
logger.debug("disabling all ONNX graph optimizations")
sess.graph_optimization_level = GraphOptimizationLevel.ORT_DISABLE_ALL sess.graph_optimization_level = GraphOptimizationLevel.ORT_DISABLE_ALL
elif "onnx-optimization-basic" in self.optimizations: elif "onnx-optimization-basic" in self.optimizations:
logger.debug("enabling basic ONNX graph optimizations")
sess.graph_optimization_level = GraphOptimizationLevel.ORT_ENABLE_BASIC sess.graph_optimization_level = GraphOptimizationLevel.ORT_ENABLE_BASIC
elif "onnx-optimization-all" in self.optimizations: elif "onnx-optimization-all" in self.optimizations:
logger.debug("enabling all ONNX graph optimizations")
sess.graph_optimization_level = GraphOptimizationLevel.ORT_ENABLE_ALL sess.graph_optimization_level = GraphOptimizationLevel.ORT_ENABLE_ALL
if "onnx-deterministic-compute" in self.optimizations: if "onnx-deterministic-compute" in self.optimizations:
logger.debug("enabling ONNX deterministic compute")
sess.use_deterministic_compute = True sess.use_deterministic_compute = True
return sess return sess