在网络内使用算子ops.ROIAlign时,出现nan的问题,主要是输出的box大小小于指定的尺寸时出现。(使用dump算子将该算子的输入输出都保存下来,单独运行该算子不会报错)
解决办法:
将该算子的默认参数都补上去,例如如下:
此时再运行,nan的问题不复见。
在网络内使用算子ops.ROIAlign时,出现nan的问题,主要是输出的box大小小于指定的尺寸时出现。(使用dump算子将该算子的输入输出都保存下来,单独运行该算子不会报错)
解决办法:
将该算子的默认参数都补上去,例如如下:
此时再运行,nan的问题不复见。
**粗体** _斜体_ [链接](http://example.com) `代码` - 列表 > 引用
。你还可以使用@
来通知其他用户。