Did you solve it? I have similar problem while using bert.
I’m following this toturial https://bugspeed.xyz/multi-label-text-classification-using-bert/
but i got ValueError: Using a target size (torch.Size([1, 4])) that is different to the input size (torch.Size([1, 5])) is deprecated. Please ensure they have the same size.