Merge pull request #3106 from dakshvar22/master

Fix Issue : Change exporter.py in object_detection

Merge pull request #3106 from dakshvar22/master
Fix Issue : Change exporter.py in object_detection
e3d7ff4a · derekjchow · GitHub · 364f96dd · 9ccb6622 · e3d7ff4a
Unverified Commit e3d7ff4a authored Jan 10, 2018 by derekjchow Committed by GitHub Jan 10, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

research/object_detection/exporter.py research/object_detection/exporter.py +1 -1

No files found.
--- 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(
-          optimize_tensor_layout=True)
+          layout_optimizer=rewriter_config_pb2.RewriterConfig.ON)
      rewrite_options.optimizers.append('pruning')
      rewrite_options.optimizers.append('constfold')
      rewrite_options.optimizers.append('layout')