Hey, I found a way to print probabilities, that is
output = model.forward(images)
prob = F.softmax(output, dim=1)
classes = Variable(torch.LongTensor(10, 1).random_(0, 3))
class_prob = torch.gather(prob, 1, classes)

I have certain concern that do I have to chance my classes?
I have also updated my post and added my actual file where I am working, to get this curve. Thanks, Sebastian.