I am including BERT in my preprocessing stage via tf.data.Dataset with a map where a BERT model loaded from tensorflow hub. This results in a very slow output. In c
tf.data.Dataset