Loss error for bert token classifier

Could you print dataset_1 to see how it looks ?