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)
以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。如有错误或未考虑完全的地方，望不吝赐教。
相关文章
这篇文章主要介绍了python 基于opencv操作摄像头的方法，帮助大家更好的理解和使用python，感兴趣的朋友可以了解下2020-12-12
这篇文章主要介绍了利用Python画双摆，绘画双摆的过程主要包括以下步骤，双摆问题、运动过程及公式推导过程，下文详细介绍，需要的小伙伴可以参考一下2022-04-04
这篇文章主要介绍利用python爬取京东商品分类以及对应的连接，这个功能不是很复杂，没有爬取里面的隐藏的东西。算是给新手一个示例教程吧，有需要的可以参考借鉴。2016-08-08
这篇文章主要介绍了python 动态绘制爱心的示例，帮助大家利用python绘制图像，感兴趣的朋友可以了解下2020-09-09
在本篇文章中小编给大家整理的是关于用Python配平化学方程式的方法以及相关注意知识点，需要的朋友们参考学习下。2019-07-07
今天小编就为大家分享一篇tensorflow 只恢复部分模型参数的实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-01-01
这篇文章主要介绍了python实现读取类别频数数据画水平条形图案例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-04-04
python虚拟环境的安装和配置(virtualenv，virtualenvwrapper）
这篇文章主要介绍了python虚拟环境的安装和配置(virtualenv，virtualenvwrapper），文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2019-08-08
今天小编就为大家分享一篇python用pandas数据加载、存储与文件格式的实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-12-12
这篇文章主要介绍了Python3+Pygame实现射击游戏完整代码，本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下2021-03-03
最新评论