better ONNX optimization logging
This commit is contained in:
parent
881b290116
commit
e0a62ccbb5
|
@ -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
|
||||||
|
|
Loading…
Reference in New Issue