CapsNet
December 8, 2017 ยท View on GitHub
This is a pytorch implementation of CapsNet, described in the paper Dynamic Routing Between Capsules - by Sara Sabour, Nicholas Frosst and Geoffrey E Hinton.
MNIST
Accuracy
Although the paper reports an accuracy of 99.75%, the maximum accuracy achieved by this implementation was 99.68%.
Execution Speed
111 seconds per epoch on a single Titan Xp GPU.
Number of Parameters
Model has 8141840 parameters.