Implemention of Capsule Net from the paper Dynamic Routing Between Capsules
Branch: master
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
.gitignore
CapsuleLayer.py update and run pass Nov 12, 2017
CapsuleNet.py
LICENSE
README.md
main.py

README.md

CapsuleNet-Gluon

Implemention of Capsule Net from the paper Dynamic Routing Between Capsules Edit Add topics

Run

  1. pip install --pre mxnet-cu80 -i https://pypi.douban.com/simple --user
  2. pip install tqdm
  3. python main.py

Results

Results to be added

Issues

I use 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.

Other Implementations