CTL RN50 uses persist BN

Author: kaixih

official/vision/image_classification/resnet_ctl_imagenet_main.py

@@ -197,6 +197,8 @@ def run(flags_obj):
         'mixed_bfloat16')
     tf.compat.v2.keras.mixed_precision.experimental.set_policy(policy)
 
+  common.set_cudnn_batchnorm_mode()
+
   # TODO(anj-s): Set data_format without using Keras.
   data_format = flags_obj.data_format
   if data_format is None: