Toy Autoencoder with multi-task learning that learns to reconstruct the input and classify the correct label on MNIST dataset.
Run the convolutional autoencoder using default hyper-parameters.
python main.py
Run the vanilla autoencoder using default hyper-parameters.
python main.py --vanilla