Group Equivariant Capsule Networks in PyTorch

Proof of concept Pytorch implementation of Group Equivariant Capsule Networks (NIPS 2018).

Required:

1. PyTorch 0.4.1
2. PyTorch Geometric
3. pytorch_unique

Install:

1. Clone Repo
2. python setup.py install

In repo dir: python examples/mnist.py to run the MNIST Experiment.

Implementations of capsules in pooling_capsule_layer.py

Implementations of sparse group convolution in group_conv_layer.py