pytorch tensor int型除法出现的问题
更新时间：2021年04月08日 14:17:53 作者：judgechen1997
这篇文章主要介绍了pytorch tensor int型除法出现的问题，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
昨天晚上跑起来一个classification实验，今天发现训练loss在降，然而accuracy永远是0 。。。直觉告诉我evaluation有问题
然后发现自己写了个很愚蠢的bug
accuracy对应的tensor出来是int型的，我用到了一个除法取平均。而pytorch里无论用 / or // 结果都是取整，，accuracy一直就没有。。
所以转换成float就没问题了，，低级bug，下不为例
补充：pytorch tensor division/除法
除法的时候要注意数据类型的问题
a = torch.div(torch.tensor([160, 110]), 0.137)
jupyter notebook就会奔溃, 因为torch.tensor([160, 110])是torch.int64数据类型, 而0.137(或者说torch.tensor(0.137))是torch.float32类型.
如果改成下面这样, 就会顺利运行
a = torch.div(torch.tensor([160, 110]).float(), 0.137)
以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。如有错误或未考虑完全的地方，望不吝赐教。
相关文章
这篇文章主要为大家详细介绍了Pygame框架实现飞机大战，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2020-08-08
要了解jinja2,那么需要先理解模板的概念。模板在Python的web开发中广泛使用，它能够有效的将业务逻辑和页面逻辑分开，使代码可读性增强，并且更加容易理解和维护。模板简单来说就是一个其中包含占位变量表示动态部分的文，模板文件在经过动态赋值后，返回给用户2023-03-03
今天小编就为大家分享一篇数据清洗--DataFrame中的空值处理方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-07-07
这篇文章主要为大家介绍了python高级搜索来高效搜索GitHub，从而高效获取所需资源，有需要的朋友可以借鉴参考下，希望能够有所帮助2021-11-11
这篇文章主要介绍了Django URL和View的关系说明,具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2021-03-03
本文主要介绍了pytest配置文件pytest.ini的具体使用，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2022-07-07
pymysql是Python3.x中操作MySQL数据库的模块，其兼容于MySQLdb，使用方法也与MySQLdb几乎相同，但是性能不如MySQLdb，但是由于其安装使用方便、对中文兼容性也更好等优点，被广泛使用。可以使用pip install pymysql进行安装。2021-09-09
浅析python打包工具distutils、setuptools
python包在开发中十分常见，一般的使用套路是所有的功能做一个python模块包，打包模块，然后发布，安装使用。这篇文章给大家介绍了python打包工具distutils、setuptools的相关知识，感兴趣的朋友一起看看吧2018-04-04
这篇文章主要介绍了Python使用pylab库实现绘制直方图功能,结合实例形式分析了Python数据读取、遍历以及基于pylab库绘制直方图的相关操作技巧,需要的朋友可以参考下2018-06-06
这篇文章主要介绍了Python中用于返回绝对值的abs()方法,是Python入门中的基础知识,需要的朋友可以参考下2015-05-05
最新评论