Custom TF model with tf.keras.layers.Embedding

Yes, exactly, I haven’t checked yet. I think there is a way to disable that.