XifengGuo / CapsNet-Keras

A Keras implementation of CapsNet in NIPS2017 paper "Dynamic Routing Between Capsules". Now test error = 0.34%.
MIT License
2.46k stars 652 forks source link

Can we run this on Theano? #22

Closed vivek14632 closed 6 years ago

vivek14632 commented 6 years ago

Hi Xi,

I want to run the code on Theano. Can you suggest any other implementation of CapsNet which is over Theano?.

Thanks.

XifengGuo commented 6 years ago

@vivek14632 I believe this repo can be easily adapted to Theano backend by taking care of the order of data channels and some tf functions. There maybe other implementation by Theano, which you can search on GitHub.

vivek14632 commented 6 years ago

Thanks @XifengGuo