Capsule Networks (PyTorch Implementation)

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

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

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

Name		Name	Last commit message	Last commit date
Latest commit History 8 Commits
layers		layers
losses		losses
models		models
.gitignore		.gitignore
README.md		README.md
main.py		main.py
tests.py		tests.py

Provide feedback