Internal change

PiperOrigin-RevId: 342937744

Internal change
PiperOrigin-RevId: 342937744
9e0cd251 · Reed Wanderman-Milne · A. Unique TensorFlower · 1d84aa30 · 9e0cd251
Commit 9e0cd251 authored Nov 17, 2020 by Reed Wanderman-Milne Committed by A. Unique TensorFlower Nov 17, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 4 deletions

official/nlp/transformer/transformer_main.py official/nlp/transformer/transformer_main.py +4 -4

No files found.
--- a/official/nlp/transformer/transformer_main.py
+++ b/official/nlp/transformer/transformer_main.py
@@ -175,9 +175,8 @@ class TransformerTask(object):
    else:
      logging.info("Not using any distribution strategy.")

-    performance.set_mixed_precision_policy(
-        params["dtype"],
-        flags_core.get_loss_scale(flags_obj, default_for_fp16="dynamic"))
+    performance.set_mixed_precision_policy(params["dtype"],
+                                           use_experimental_api=False)

  @property
  def use_tpu(self):
@@ -443,7 +442,8 @@ class TransformerTask(object):
        use_float16=params["dtype"] == tf.float16,
        use_graph_rewrite=self.flags_obj.fp16_implementation == "graph_rewrite",
        loss_scale=flags_core.get_loss_scale(
-            self.flags_obj, default_for_fp16="dynamic"))
+            self.flags_obj, default_for_fp16="dynamic"),
+        use_experimental_api=False)

    return opt