Yeah but I think my labels are fine, like they start from zero as they should in a BERT model, here have a look at my dataset as a dataframe object
I think the problem lies in assigning the num_labels in the model from the config file of model
And now I am getting a new error
ValueError: Target size (torch.Size([1])) must be the same as input size (torch.Size([1, 2]))