I am new to Hugging face models and deep learning in general. I search online and found that the above error occurs because I am getting 9 columns as prediction where as I should have just one. The exact error is: File "train.py", line 107, in <module> trainer.train() File "/opt/conda/lib/pyt…

Target size (torch.size([16])) must be the same as input size (torch.size([16, 9]))

sana-ngu December 20, 2022, 3:58pm 2

I am having the same error, did you manage to solve it?

Topic		Replies	Views
ValueError: Target size (torch.Size([8])) must be the same as input size (torch.Size([8, 8])) Beginners	15	12621	January 9, 2025
Target size (torch.Size([8])) must be the same as input size (torch.Size([8, 2])) 🤗Transformers	5	5545	October 13, 2023
Mismatched target and input size for BCE using "multi_label_classification" Intermediate	2	7025	September 1, 2022
ValueError: Target size (torch.Size([8])) must be the same as input size (torch.Size([8, 3])) Beginners	0	634	December 20, 2022
Problem "Target size must be the same as input size " 🤗Transformers	0	306	March 19, 2024