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