I am trying to train a simple CNN on custom image data. While training with raw numpy data, the model learns as expected. However when I convert the same raw numpy data to dataset using tf.data.Dataset.from_tensor_slices, the training goes completely goes haywire, and the model does not learn anything.