Skip to content

GitLab

Commit 0d1fb4cf authored by Kaixi Hou's avatar Kaixi Hou
Browse files

Add more comments

parent a5304312
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
official/vision/image_classification/resnet_ctl_imagenet_main.py
View file @ 0d1fb4cf
...@@ -197,6 +197,7 @@ def run(flags_obj): ...@@ -197,6 +197,7 @@ def run(flags_obj):
'mixed_bfloat16') 'mixed_bfloat16')
tf.compat.v2.keras.mixed_precision.experimental.set_policy(policy) tf.compat.v2.keras.mixed_precision.experimental.set_policy(policy)
# This only affects GPU.
common.set_cudnn_batchnorm_mode() common.set_cudnn_batchnorm_mode()
# TODO(anj-s): Set data_format without using Keras. # TODO(anj-s): Set data_format without using Keras.
......
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