Skip to content

GitLab

Unverified Commit 9ccb6622 authored by Daksh Varshneya's avatar Daksh Varshneya Committed by GitHub
Browse files

Make parameter value more generic and adaptable. 

Incorporated the suggested changes.
parent da98beae
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
research/object_detection/exporter.py
View file @ 9ccb6622
...@@ -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(
layout_optimizer=1) layout_optimizer=rewriter_config_pb2.RewriterConfig.ON)
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