Thanks to visit codestin.com
Credit goes to github.com

Skip to content

Commit da98bea

Browse files
committed
Change in Object Dectection Model. Change parameter name given to RewriterConfig Proto. optimize_tensor_layout was changed to layout_optimizer in current tf version.
1 parent ad5c1fb commit da98bea

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

research/object_detection/exporter.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -69,7 +69,7 @@ def freeze_graph_with_def_protos(
6969
if optimize_graph:
7070
logging.info('Graph Rewriter optimizations enabled')
7171
rewrite_options = rewriter_config_pb2.RewriterConfig(
72-
optimize_tensor_layout=True)
72+
layout_optimizer=1)
7373
rewrite_options.optimizers.append('pruning')
7474
rewrite_options.optimizers.append('constfold')
7575
rewrite_options.optimizers.append('layout')

0 commit comments

Comments
 (0)