Extending the Capslayers

[pgarz]

Hello! I'm currently working on a project where I'd like to experiment with capsules in lieu of CNNs for Deep-Q Learning. Great work on releasing this implementation! While working with this code I ran into issues with using more capsule layers than just the ones in the CapsNet architecture. For instance, I was wondering if was possible to use multiple convolutional capsule layers with routing and to change their output sizes? I've tried to tweak the code to do this but I keep running into size issues and fear I might break the logical implementation. Any tips greatly appreciated!