pytorch张量的.item()和.numpy()
今天在加载一个pytorch代码时出现了在测试集上的精度计算为0的情况。于是小白我又懵圈了,代码明明是按书上敲的,怎么就是不对呢。代码本身非常简单,是一个MNIST数据集上的hello world代码。但没想到后面在测试集上的精度计算却给我来了一个意外的“惊喜”。(后面有MNIST的整个代码,和我写的精度为零的部分。),经过参考https://www.jianshu.com/p/be3276b434b2文章上的相关内容,我也只是知道了一个大概的修改方法,并在最后留下了一个疑问,至于为什么会这样,还是希望哪位路过的大神给解释一下。
问题描述:
在使用最后的代码进行求解精度的时候,精度的计算为0。出错代码行:
print('Accuracy of the network on the 10000 test images: %d %%'%(100*(correct/total)))
输出的内容为:
解决办法:
经过分别打印correct和total的类型,发现correct是tensor类型,而total是int类型,因此想