Commit d5722dcd authored by Vinh Nguyen's avatar Vinh Nguyen
Browse files

minor fix to comments

parent c9c05e9b
...@@ -380,7 +380,7 @@ class TransformerTask(object): ...@@ -380,7 +380,7 @@ class TransformerTask(object):
# Note: when flags_obj.fp16_implementation == "graph_rewrite", # Note: when flags_obj.fp16_implementation == "graph_rewrite",
# dtype as determined by flags_core.get_tf_dtype(flags_obj) would be 'float32' # dtype as determined by flags_core.get_tf_dtype(flags_obj) would be 'float32'
# which will ensure tf.keras.mixed_precision and tf.train.experimental.enable_mixed_precision_graph_rewrite # which will ensure tf.keras.mixed_precision and tf.train.experimental.enable_mixed_precision_graph_rewrite
# does not double up. # do not double up.
opt = tf.train.experimental.enable_mixed_precision_graph_rewrite(opt) opt = tf.train.experimental.enable_mixed_precision_graph_rewrite(opt)
return opt return opt
......
...@@ -188,7 +188,7 @@ def run(flags_obj): ...@@ -188,7 +188,7 @@ def run(flags_obj):
# Note: when flags_obj.fp16_implementation == "graph_rewrite", # Note: when flags_obj.fp16_implementation == "graph_rewrite",
# dtype as determined by flags_core.get_tf_dtype(flags_obj) would be 'float32' # dtype as determined by flags_core.get_tf_dtype(flags_obj) would be 'float32'
# which will ensure tf.keras.mixed_precision and tf.train.experimental.enable_mixed_precision_graph_rewrite # which will ensure tf.keras.mixed_precision and tf.train.experimental.enable_mixed_precision_graph_rewrite
# does not double up. # do not double up.
optimizer = tf.train.experimental.enable_mixed_precision_graph_rewrite(optimizer) optimizer = tf.train.experimental.enable_mixed_precision_graph_rewrite(optimizer)
if flags_obj.use_trivial_model: if flags_obj.use_trivial_model:
......
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