Run the program:
python3 autoencoder.py -d ./Datasets/t10k-images-idx3-ubyte
python3 classification.py -d Datasets/train-images-idx3-ubyte -dl Datasets/train-labels-idx1-ubyte -t Datasets/t10k-images-idx3-ubyte -tl Datasets/t10k-labels-idx1-ubyte -model model.h5
If you want to access the code in colab, click the link above. In colab, the TPU option is enabled for fast performance