Skip to content

GitLab

Commit a5304312 authored by kaixih's avatar kaixih
Browse files

CTL RN50 uses persist BN

parent 8d9a16ce
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
official/vision/image_classification/resnet_ctl_imagenet_main.py
View file @ a5304312
...@@ -197,6 +197,8 @@ def run(flags_obj): ...@@ -197,6 +197,8 @@ 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)
common.set_cudnn_batchnorm_mode()
# TODO(anj-s): Set data_format without using Keras. # TODO(anj-s): Set data_format without using Keras.
data_format = flags_obj.data_format data_format = flags_obj.data_format
if data_format is None: if data_format is None:
......
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