This is a pytorch implementation of CapsNet, described in the paper Dynamic Routing Between Capsules - by Sara Sabour, Nicholas Frosst and Geoffrey E Hinton.

Although the paper reports an accuracy of 99.75%, the maximum accuracy achieved by this implementation was 99.68%.

111 seconds per epoch on a single Titan Xp GPU.

Model has 8141840 parameters.