1. Toy模板网首页
  2. > Toy博客

PyTorch训练“WARNING:root:NaN or Inf found in input tensor”问题

9月前 作者：ya6543 分类：Toy博客 阅读(41) 违法举报

这篇具有很好参考价值的文章主要介绍了PyTorch训练“WARNING:root:NaN or Inf found in input tensor”问题。希望对大家有所帮助。如果存在错误或未考虑完全的地方，请大家不吝赐教，您也可以点击"举报违法"按钮提交疑问。

问题

使用PyTorch训练开始时报以下警告：

WARNING:root:NaN or Inf found in input tensor

训练中也会偶尔再报同样的警告，但是似乎不影响正常训练。

分析

查了一下其他人也有报这个警告的情况，一般的解释都是模型训练过程中出现梯度消失或者梯度爆炸，或者数据中或数据处理过程中出现了脏数据。但是经过调试，发现不管是改大改小学习率都仍然出现此警告，抓取脏数据也抓取不到，在数据集的构造、网络的计算中也都没有发现脏数据。

分步调试后发现该警告出现在第一个epoch结束之后，即train和validation的正向和反向传播都完成之后。怀疑是在tensorboardX使用中的问题（这个issue也被怀疑是tensorboard的问题：NaN or Inf found in input tensor #223），于是仔细检查了tensorboardX语句中可能出现问题的部分。发现有一处add_image中输入的image值超过了1，于是将这一句注释掉再运行，果然不报警告了。

结论

除了模型训练过程中出现梯度消失或者梯度爆炸，或者数据中或数据处理过程中出现了脏数据，还有可能是tensorboardX使用中数据格式不符合规范。文章来源地址https://www.toymoban.com/news/detail-540770.html

到了这里，关于PyTorch训练“WARNING:root:NaN or Inf found in input tensor”问题的文章就介绍完了。如果您还想了解更多内容，请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章，希望大家以后多多支持TOY模板网！

本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如若转载，请注明出处： 如若内容造成侵权/违法违规/事实不符，请点击违法举报进行投诉反馈，一经查实，立即删除！

分享到：
领支付宝红包 赞助服务器费用

相关文章

觉得文章有用就打赏一下文章作者

支付宝扫一扫打赏

博客赞助

微信扫一扫打赏

请作者喝杯咖啡吧~博客赞助

支付宝扫一扫领取红包，优惠每天领

二维码1

领取红包

二维码2

领红包