ValueError: Target size (torch.Size([8])) must be the same as input size (torch.Size([8, 3]))

Beginners
1

I am trying to do multi label classification
I am getting this error :

Screenshot 2022-12-20 at 16.54.27
Screenshot 2022-12-20 at 16.54.271741×1151 357 KB

any help?
I am using this model =AutoModelForSequenceClassification.from_pretrained(checkpoint,problem_type=‘multi_label_classification’,num_labels=3).to(device)