@ptrblck @7029279
i just solved error but unfortunately my models are predicting class like 0,1,2,3,4 but i wanted it to generate confident score for each image like 0.7345, 0.48045 etc
i think it’s because of the loss function i am using,is there anything wrong in my code? how can i get this model to generate confident score? should i try MSEloss or crossentropy?  but if i use crossentropy and do loss = criterion(outputs, torch.max(labels, 1)[1]) then i get loss = 0.000 any help?