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.