Skip to content

GitLab

"examples/hubert/vscode:/vscode.git/clone" did not exist on "4c19e2cb3ef78ab3f6ae4052182c3aeed4817d08"
Commit c186c85a authored by Vinh Nguyen's avatar Vinh Nguyen
Browse files

change default fp16_implementation to graph_rewrite

parent 97c19f54
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/utils/flags/_performance.py
View file @ c186c85a
...@@ -192,7 +192,7 @@ def define_performance(num_parallel_calls=True, inter_op=True, intra_op=True, ...@@ -192,7 +192,7 @@ def define_performance(num_parallel_calls=True, inter_op=True, intra_op=True,
if fp16_implementation: if fp16_implementation:
# Currently, this flag is only defined for the estimator resnet model. # Currently, this flag is only defined for the estimator resnet model.
flags.DEFINE_enum( flags.DEFINE_enum(
name="fp16_implementation", default="casting", name="fp16_implementation", default="graph_rewrite",
enum_values=("casting', 'graph_rewrite"), enum_values=("casting', 'graph_rewrite"),
help=help_wrap( help=help_wrap(
"When --dtype=fp16, how fp16 should be implemented. This has no " "When --dtype=fp16, how fp16 should be implemented. This has no "
......
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