search

XifengGuo / CapsNet-Keras

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

Batch size built into the model [tf2 branch] #111

Open bklooste opened 4 years ago

bklooste commented 4 years ago

The batch size if built into the model , this creates quite a few issues and can fail when dealing with different training sizes or filtering data .

CoffeeStraw commented 4 years ago

Have you managed to solve that?

essamgoda commented 4 years ago

I have Same error.

bklooste commented 3 years ago

Its a fair bit of work due to the way it was done .. almost need to redo .

CoffeeStraw commented 3 years ago

I think I've managed to make this work... Unfortunately, I've organized my codebase differently from Xifeng implementation (even if my work is based on its implementation) and I will not be able to prepare a PR for this project in a while. For those interested, here's my work:

https://github.com/CoffeeStraw/CapsNet-Knowledge-Extractor/blob/master/capsnet_trainer/_share/capslayers.py