Skip to content

Latest commit

 

History

History
9 lines (6 loc) · 459 Bytes

README.md

File metadata and controls

9 lines (6 loc) · 459 Bytes

Capsule Networks (PyTorch Implementation)

PyTorch implementation of Dynamic Routing Between Capsules[1]. This repository includes experiments on MNIST, FashionMNIST, and CIFAR10.

Credits

Some general ideas of implementation have been borrowed from gram-ai/capsule-networks.

References

[1] Sara Sabour, Nicholas Frosst, Geoffrey E Hinton. Dynamic Routing Between Capsules, NIPS 2017.