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 2) 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