How to handle increased training with tf.keras.layers.Embedding layer?

Question

I have a model that encodes item id into the embedding vector in the training process. I\'m using tf.keras.layers.Embedding to do this. However, when the model is t