Make parameter value more generic and adaptable.

dakshvar22 · web-flow · commit 9ccb66226642 · 2018-01-10T13:20:06.000+05:30
Incorporated the suggested changes.
diff --git a/research/object_detection/exporter.py b/research/object_detection/exporter.py
@@ -69,7 +69,7 @@ def freeze_graph_with_def_protos(
     if optimize_graph:
       logging.info('Graph Rewriter optimizations enabled')
       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('constfold')
       rewrite_options.optimizers.append('layout')
