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.