Closed glcanvas closed 2 years ago
I think the error message is self-explained. Your label shape [20] doesn't look right. It should be [4,5]. 4 is the batch size and 5 is the 5-class vector. You can either reshape your label or flatten your model prediction
yes, I know that's self-explained but this is your code=) I launched it without any modification
OK, thanks for reporting this issue. I will look into it.
you may fix it in 2 lines =)
x_out_empty_ind1 = torch.squeeze(x_out_empty_ind1)
x_out_empty_ind1 = torch.reshape(x_out_empty_ind1, (1, x_out_empty_ind1.shape[0]))
same for x_out_empty_ind2
Hi! I got error while launch train.py script
my command