当前位置: 代码迷 >> 综合 >> 解决block: [0,0,0], thread: [11,0,0] Assertion `input_val >= zero input_val <= one` failed 问题
  详细解决方案

解决block: [0,0,0], thread: [11,0,0] Assertion `input_val >= zero input_val <= one` failed 问题

热度:86   发布时间:2023-12-14 09:48:40.0

今天在用 GPU 跑模型时,报了如下错误,绝望,简直是绝望。。。
在这里插入图片描述
错误定位到了计算 loss 的位置,通过打日志,记录传入loss的label和预测值,发现出错时,预测值里有 nan
在这里插入图片描述
检查代码,发现了除 0 运算,改之即可。

查阅资料发现,该错误还有可能由以下原因引起:

  • 预测值和标签长度不一致
  • 预测值或标签值不在 [0, 1] 范围内
  相关解决方案