Tensorflow - Keras: Consider either turning off auto-sharding or switching the auto_shard_policy to DATA to shard this dataset

Question

While training a model in keras / tensorflow:

The code snippet:

strategy = tf.distribute.experimental.MultiWorkerMirroredStrategy()

I