fb35d6bef6eecc640fb865dd2fc73d8fee2a93b6,official/vision/image_classification/resnet_ctl_imagenet_main.py,,run,#Any#,98

Before Change

      enable_xla=flags_obj.enable_xla)

  dtype = flags_core.get_tf_dtype(flags_obj)
  if dtype == tf.float16:
    policy = tf.compat.v2.keras.mixed_precision.experimental.Policy(
        "mixed_float16")
    tf.compat.v2.keras.mixed_precision.experimental.set_policy(policy)
  elif dtype == tf.bfloat16:
    policy = tf.compat.v2.keras.mixed_precision.experimental.Policy(
        "mixed_bfloat16")
    tf.compat.v2.keras.mixed_precision.experimental.set_policy(policy)

  // This only affects GPU.
  common.set_cudnn_batchnorm_mode()

  // TODO(anj-s): Set data_format without using Keras.
  data_format = flags_obj.data_format

After Change

  keras_utils.set_session_config(
      enable_eager=flags_obj.enable_eager,
      enable_xla=flags_obj.enable_xla)
  performance.set_mixed_precision_policy(flags_core.get_tf_dtype(flags_obj))

  // This only affects GPU.
  common.set_cudnn_batchnorm_mode()

Instances