关于神经网络:opsROIAlgin算子出现nan的问题解决

在网络内应用算子ops.ROIAlign时,呈现nan的问题,次要是输入的box大小小于指定的尺寸时呈现。(应用dump算子将该算子的输入输出都保留下来,独自运行该算子不会报错)

解决办法:

将该算子的默认参数都补上去,例如如下:

此时再运行,nan的问题不复见。

评论

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注

这个站点使用 Akismet 来减少垃圾评论。了解你的评论数据如何被处理