I was training a simple neural network with the Fashion MNIST dataset which is directly available in the tf.keras datasets API. I trained the model in Colab and IBM Watson S