Skip to content

GitLab

"test/git@developer.sourcefind.cn:change/sglang.git" did not exist on "a6ea3add768ff4e26ad1c40b06b49ca75f41a574"
Commit 5e8a51fd authored by Steven Hickson's avatar Steven Hickson Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 353260280
parent 3efdd18b
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/modeling/performance.py
View file @ 5e8a51fd
...@@ -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