Implemention of Capsule Net from the paper Dynamic Routing Between Capsules Edit Add topics
- pip install --pre mxnet-cu80 -i https://pypi.douban.com/simple --user
- pip install tqdm
- python main.py
Results to be added
SGD to try Capsule Net, if use
Adam as described in the original paper, I got very low test accuracy.
I find if use large train batch_size, the results is also unaceptable. I haven't figure it out.
Any PR is welcomed.
Kaggle (this version as self-contained notebook):