0xpantera/ai_lab_gt_capsNet

PyTorch Implementation of Capsule Networks on MNIST

PyTorch Implementation of Capsule Networks on MNIST

Based on the paper Dynamic Routing Between Capsules¹

Conda Environment

To reproduce the results install the conda environment from the spec-file.txt. by running:

conda install --name capsnet --file spec-file.txt

Footnotes

1. Sara Sabour, Nicholas Frosst, Geoffrey E Hinton ↩