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 652 forks source link

Increase the training/test speed of CaspNet? #97

Open Garfield2013 opened 5 years ago

Garfield2013 commented 5 years ago
  1. What ways exists for optimizing the speed of the training/testing of CapsNet?

  2. Can Nvidia's TensorRT framework optimize the interference of CapsNet?

  3. Would CapsNet with EM routing help anything in this regard?