ad09cf49a8ec2757c475002909e029d1a9ade14c,official/vision/image_classification/resnet/resnet_imagenet_main.py,,run,#Any#,41

Before Change

  data_format = flags_obj.data_format
  if data_format is None:
    data_format = ("channels_first"
                   if tf.test.is_built_with_cuda() else "channels_last")
  tf.keras.backend.set_image_data_format(data_format)

  // Configures cluster spec for distribution strategy.

After Change

  data_format = flags_obj.data_format
  if data_format is None:
    data_format = ("channels_first" if tf.config.list_physical_devices("GPU")
                   else "channels_last")
  tf.keras.backend.set_image_data_format(data_format)

  // Configures cluster spec for distribution strategy.

Instances