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

Internal change

PiperOrigin-RevId: 353260280
parent 6d4ba79d
......@@ -40,7 +40,7 @@ def configure_optimizer(optimizer,
optimizer, dynamic=False, initial_scale=loss_scale)
if use_graph_rewrite:
# 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
# up.
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