Chainer implementation of CapsNet for MNIST.
For the detail, see Dynamic Routing Between Capsules, Sara Sabour, Nicholas Frosst, Geoffrey E Hinton, NIPS 2017.
python -u train.py -g 0
Test accuracy of a trained model reached 0.9959.
TODO
- Add reconstruction loss
- Find better hyperparameters