Skip to content

GitLab

Commit 83145027 authored by Steven Hickson's avatar Steven Hickson Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 353260280
parent 6d4ba79d
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/modeling/performance.py
View file @ 83145027
...@@ -40,7 +40,7 @@ def configure_optimizer(optimizer, ...@@ -40,7 +40,7 @@ def configure_optimizer(optimizer,
optimizer, dynamic=False, initial_scale=loss_scale) optimizer, dynamic=False, initial_scale=loss_scale)
if use_graph_rewrite: if use_graph_rewrite:
# Note: the model dtype must be 'float32', which will ensure # Note: the model dtype must be 'float32', which will ensure
# tf.ckeras.mixed_precision and # tf.keras.mixed_precision and
# tf.train.experimental.enable_mixed_precision_graph_rewrite do not double # tf.train.experimental.enable_mixed_precision_graph_rewrite do not double
# up. # up.
optimizer = tf.train.experimental.enable_mixed_precision_graph_rewrite( optimizer = tf.train.experimental.enable_mixed_precision_graph_rewrite(
......
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