@SherlockLiao
Hmm, I don’t seem to get the same topk_prob as you:
For example, this is my top5_prob printout:
Variable containing:
 3.6295 -0.3418 -1.3023 -2.8497 -2.8511
[torch.cuda.FloatTensor of size 1x5 (GPU 0)]

Not sure what these values are and how I can convert the values to probabilities.