An error occurs when predict with the same data as when performing train (expects 3 input(s), but it received 25 input tensors.)

Question

The model was constructed using electra.

here is my model

input_ids = tf.keras.Input(shape=(MAX_LEN,), name=\'input_ids\', dtype=tf.in