Skip to content

GitLab

Commit 4c2f424f authored by Reed Wanderman-Milne's avatar Reed Wanderman-Milne Committed by A. Unique TensorFlower
Browse files

Pass use_experimental_api=True to mixed precision functions. 

The default is True, but I plan on changing it to False soon. After that, I plan on removing the argument and never using the experimental API.

PiperOrigin-RevId: 360724698
parent c9ab5a7a
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
official/benchmark/models/resnet_imagenet_main.py
View file @ 4c2f424f
...@@ -92,7 +92,8 @@ def run(flags_obj): ...@@ -92,7 +92,8 @@ def run(flags_obj):
dtype = flags_core.get_tf_dtype(flags_obj) dtype = flags_core.get_tf_dtype(flags_obj)
performance.set_mixed_precision_policy( performance.set_mixed_precision_policy(
flags_core.get_tf_dtype(flags_obj), flags_core.get_tf_dtype(flags_obj),
flags_core.get_loss_scale(flags_obj, default_for_fp16=128)) flags_core.get_loss_scale(flags_obj, default_for_fp16=128),
use_experimental_api=True)
data_format = flags_obj.data_format data_format = flags_obj.data_format
if data_format is None: if data_format is None:
......
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