stacked-capsule-networks

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.

Name		Name	Last commit message	Last commit date
Latest commit History 30 Commits
data/MNIST		data/MNIST
.gitignore		.gitignore
README.md		README.md
main.py		main.py
model.py		model.py
setmodules.py		setmodules.py

Provide feedback