I\'m using TFBertForTokenClassification to perform a NER task on the annotated corpus fo NER: https://www.kaggle.com/abhinavwalia95/entity-annotated-corpus. The problem is t