IOPub data rate exceeded using RoBERTa

Question

I\'m trying to set up a roBERTa model on tensorflow for classification but have an issue when I try to fit the model.

Here is the set up of the model.