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)
以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。如有错误或未考虑完全的地方，望不吝赐教。
相关文章
Django配置celery（非djcelery）执行异步任务和定时任务
这篇文章主要介绍了Django配置celery（非djcelery）执行异步任务和定时任务，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧2018-07-07
这篇文章主要介绍了如何利用python执行txt文件中的代码，python这么强大的语言当然可以做大，只需使用内置的exex()函数，进入主题前我们先来看看什么是exec()函数，需要的小伙伴可以参考一下2022-03-03
这篇文章主要介绍了python实现的希尔排序算法,实例分析了基于Python实现希尔排序的相关技巧,需要的朋友可以参考下2015-07-07
本文主要介绍了Django数据映射(一对一,一对多,多对多)，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2022-08-08
本文主要介绍了Pycharm搭建一个Django项目的方法步骤，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2023-02-02
时间戳是一个表示日期和时间的数值,通常以秒为单位,在Python中,获取时间戳是常见的任务,用于记录事件、计时操作、以及在各种应用中跟踪时间,本文将介绍多种获取时间戳的方法,包括标准库和第三方库的方式,并提供示例代码以帮助你更好地理解2023-11-11
在Python中列表是一种非常灵活和强大的数据结构,支持多种运算和操作,这篇文章主要介绍了Python中列表乘法和列表推导式区别的相关资料,文中通过代码就介绍的非常详细,需要的朋友可以参考下2025-04-04
这篇文章主要介绍了Django权限系统auth模块用法解读，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2023-05-05
基于python requests selenium爬取excel vba过程解析
这篇文章主要介绍了基于python requests selenium爬取excel vba过程解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-08-08
这篇文章主要介绍了python正则表达式的懒惰匹配和贪婪匹配说明，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-07-07
最新评论