bert-toxic-comments-multilabel icon indicating copy to clipboard operation
bert-toxic-comments-multilabel copied to clipboard

Published 20 hours ago • verified publisher icon kaushaltrivedi

Target size is not matching with the input size

Open ramakth1 opened this issue 5 years ago • 1 comments

I have loaded the data (Toxic dataset) and tried to run the model using batch_size_per_gpu = 4 but i am getting the below error.

ValueError: Target size (torch.Size([0, 6])) must be the same as input size (torch.Size([4, 6]))

Could you please help here.

ramakth1 avatar Oct 15 '19 09:10 ramakth1

same issue

ii35322 avatar Oct 15 '19 18:10 ii35322