Here is the implementation: https://github.com/maxi-marufo/my-challenges/blob/master/IMAGE_CLASSIFICATION_Fashion_MNIST/notebook.ipynb