theailearner / capsule-network Goto Github PK

ValueError Traceback (most recent call last)
in ()
33 return K.sqrt(K.sum(K.square(inputs), -1) + K.epsilon())
34
---> 35 digit_caps = DigitCapsuleLayer()(squashed_output)

2 frames
/usr/local/lib/python3.6/dist-packages/keras/backend/tensorflow_backend.py in batch_dot(x, y, axes)
1259 str(x_shape) + ' and ' + str(y_shape) +
1260 ' with axes=' + str(axes) + '. x.shape[%d] != '
-> 1261 'y.shape[%d] (%d != %d).' % (axes[0], axes[1], d1, d2))
1262
1263 # backup ndims. Need them later.

ValueError: Can not do batch_dot on inputs with shapes (None, 10, 10, 1152, 16) and (None, 10, 1152, 1152, 16) with axes=[2, 3]. x.shape[2] != y.shape[3] (10 != 1152).