Commit da98beae authored by Daksh Varshneya's avatar Daksh Varshneya
Browse files

Change in Object Dectection Model. Change parameter name given to...

Change in Object Dectection Model. Change parameter name given to RewriterConfig Proto. optimize_tensor_layout was changed to layout_optimizer in current tf version.
parent ad5c1fbb
...@@ -69,7 +69,7 @@ def freeze_graph_with_def_protos( ...@@ -69,7 +69,7 @@ def freeze_graph_with_def_protos(
if optimize_graph: if optimize_graph:
logging.info('Graph Rewriter optimizations enabled') logging.info('Graph Rewriter optimizations enabled')
rewrite_options = rewriter_config_pb2.RewriterConfig( rewrite_options = rewriter_config_pb2.RewriterConfig(
optimize_tensor_layout=True) layout_optimizer=1)
rewrite_options.optimizers.append('pruning') rewrite_options.optimizers.append('pruning')
rewrite_options.optimizers.append('constfold') rewrite_options.optimizers.append('constfold')
rewrite_options.optimizers.append('layout') rewrite_options.optimizers.append('layout')
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment