titu1994 / keras-squeeze-excite-network

Implementation of Squeeze and Excitation Networks in Keras
MIT License
400 stars 118 forks source link

Element wise multiplication isn't working #25

Closed StasyanOi closed 3 years ago

StasyanOi commented 3 years ago

Hello, the element wise multiplication (model=multiply([init, se])) throws this error. Should the dimensions of se and init be always the same?

ValueError: Only tensors of same shape can be merged by layermultiply_1 Got input shapes: [(None, 128, 128, 64), (None, 1, 1, 64)]