Pytorch Implementation of the paper StackedCapsuleAutoEncoders. link

• The code currently supports MNIST dataset only
• The code for the set transformers part (setmodules.py) is used from the official repository [https://github.com/juho-lee/set_transformer].
• Feel free to cite this repository if you want to use the code.
• To run the code with default parameters according to the paper on MNIST dataset use python main.py