How to solve index error while fine-tuning BERT model on custom dataset

[vinrok]

Hi,

I am facing Index Error, while fine tuning BERT model on custom text corpus.

Kindly help me to solve it.

Kindly find the screenshot of error attached and colab link for code here - Google Colab

Regards,

Vinay