Hi, I’ve been having this error with the class Translator,
InvalidArgumentError: Exception encountered when calling layer ‘key’ (type EinsumDense).
cannot compute Einsum as input #1(zero-based) was expected to be a int64 tensor but is a float tensor [Op:Einsum] name:
Call arguments received by layer ‘key’ (type EinsumDense):
inputs=tf.Tensor(shape=(64, 18), dtype=int64)
Here is the full error:
I tried casting the context and target to float32 but it just lead to more errors. All the previous test were passed, so I’m not really sure what to do here
Thank you in advance