Capsule Networks - Githubissues

ShigekiKarita / grain

autograd mir and CUDA library for dynamic neural networks in D.

https://shigekikarita.github.io/grain

Boost Software License 1.0

66 stars 7 forks source link

Open nordlow opened 6 years ago

nordlow commented 6 years ago

Are you planning on adding support for capsule networks?

ShigekiKarita commented 6 years ago

Thanks for your request. I've never read the paper yet but just check out pytorch implentation https://github.com/timomernick/pytorch-capsule There are several missing ops in grain

binary ops: max
reductions ops: mean, sum, max
topology ops: slice, stack, transpose, cat, view (these are relatively harder to implement) To support topology ops, I gonna make huge changes in variable struct. Then the method implementation can be done.

nordlow commented 6 years ago

Nice. Thanks for your fast answer.