Something wrong in the top5 accuracy

[whaohan]

There may be something wrong in top5 accuracy updating, which making the top5 accuracy is the same as the top1.
The code with problem(main.py) is as follow:

[tomorrow-rain]

There may be something wrong in top5 accuracy updating, which making the top5 accuracy is the same as the top1. The code with problem(main.py) is as follow:

top5.update(prec5[0], input.size(0))