RuntimeError:invalid argument 5:k not in range for dimension at /pytorch/ate ...
这个问题是我在使用pytorch训练模型时遇到的,意思是5这个参数是无效的,我一开始认为是我数据集的问题,应该不是图片通道的问题,因为我用的都是三通道的,应该是图片尺寸不统一,就想着把它们都改成同样大小。在我改着的时候我突然意识到是topk出错了,一般训练模型时会返回top1和top5精度,top1是你预测的label取最后概率向量里面最大的那一个作为预测结果,如过你的预测结果中概率最大的那个分类正确,则预测正确。否则预测错误;top5是最后概率向量最大的前五名中,只要出现了正确概率即为预测正确。否则预测错误。
我的数据集只分了三类,根本没有top5,于是我把5都改成了3。终于开始跑了。