Loss error for bert token classifier