Skip to content

GitLab

"docs/git@developer.sourcefind.cn:OpenDAS/mmdetection3d.git" did not exist on "92d5a87e97b058621fa55a6f073a287e5ecd1a27"
Unverified Commit e3d7ff4a authored by derekjchow's avatar derekjchow Committed by GitHub
Browse files

Merge pull request #3106 from dakshvar22/master 

Fix Issue : Change exporter.py in object_detection
parents 364f96dd 9ccb6622
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
research/object_detection/exporter.py
View file @ e3d7ff4a
...@@ -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=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