My attempt at implementing CapsNet from the paper Dynamic Routing Between Capsules
Switch branches/tags
Nothing to show
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Failed to load latest commit information.
LICENSE
README.md
capsnet.py

README.md

CapsNet-PyTorch

My attempt at implementing CapsNet from the paper Dynamic Routing Between Capsules.
Link to paper
Authors of paper - Sara Sabour, Nicholas Frosst, Geoffrey E Hinton

The code is buggy right now, reconstruction loss is yet to be added.
Training-Testing not done till now.
Suggestions and contributions welcome.