KFrank:

Your labels tensor should be a pytorch tensor (not a numpy array),
and should be a batch of sets of (floating-point) binary class labels,
one for each of your 15 classes, and therefore should have shape
[250, 15] .


(fc): Linear(in_features=512, out_features=15, bias=True)




So do I need to change the last year out_features to  [250, 15] or something else?