Skip to content

GitLab

Commit 78047d54 authored by Kaixi Hou's avatar Kaixi Hou
Browse files

fixed a bad indent

parent fcb8590b
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/resnet/ctl/ctl_imagenet_main.py
View file @ 78047d54
...@@ -177,7 +177,7 @@ def run(flags_obj): ...@@ -177,7 +177,7 @@ def run(flags_obj):
"--use_tf_function to be true") "--use_tf_function to be true")
loss_scale = flags_core.get_loss_scale(flags_obj, default_for_fp16=128) loss_scale = flags_core.get_loss_scale(flags_obj, default_for_fp16=128)
optimizer = tf.train.experimental.enable_mixed_precision_graph_rewrite( optimizer = tf.train.experimental.enable_mixed_precision_graph_rewrite(
optimizer, loss_scale) optimizer, loss_scale)
training_accuracy = tf.keras.metrics.SparseCategoricalAccuracy( training_accuracy = tf.keras.metrics.SparseCategoricalAccuracy(
'training_accuracy', dtype=tf.float32) 'training_accuracy', dtype=tf.float32)
......
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