search

brjathu / deepcaps

Official Implementation of "DeepCaps: Going Deeper with Capsule Networks" paper (CVPR 2019).
MIT License
151 stars 48 forks source link

Batch dot reimplementation #16

Closed mrazekv closed 4 years ago

mrazekv commented 4 years ago

Pull request regarding to issue #14 Keras (or TF?) in the new version changed the implementation on batch_dot function. I copied the batch_dot function from TF1.13 and replaced the original version.

The code is now working even with following Anaconda modules

keras                     2.3.1                         0
keras-applications        1.0.8                      py_0
keras-base                2.3.1                    py37_0
keras-preprocessing       1.1.0                      py_1
tensorflow                1.15.0          gpu_py37h0f0df58_0
tensorflow-base           1.15.0          gpu_py37h9dcbed7_0
tensorflow-estimator      1.15.1             pyh2649769_0
tensorflow-gpu            1.15.0               h0d30ee6_0
......