/CapsNet-PyTorch

My attempt at implementing CapsNet from the paper Dynamic Routing Between Capsules

Primary LanguagePythonMIT LicenseMIT

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.