pytorch-capsule

A Pytorch implementation of Hinton's "Dynamic Routing Between Capsules".

https://arxiv.org/pdf/1710.09829.pdf

Thanks to @naturomics for his Tensorflow implementation which was a useful guide and sanity check. https://github.com/naturomics/CapsNet-Tensorflow

To use:

$ python main.py