pytorch tensor int型除法出现的问题
短信预约 -IT技能 免费直播动态提醒
昨天晚上跑起来一个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)以上为个人经验，希望能给大家一个参考，也希望大家多多支持编程网。如有错误或未考虑完全的地方，望不吝赐教。
免责声明：
① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的，并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据，供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。
② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341
