在网络内应用算子 ops.ROIAlign 时,呈现 nan 的问题,次要是输入的 box 大小小于指定的尺寸时呈现。(应用 dump 算子将该算子的输入输出都保留下来,独自运行该算子不会报错)
解决办法:
将该算子的默认参数都补上去,例如如下:
此时再运行,nan 的问题不复见。
在网络内应用算子 ops.ROIAlign 时,呈现 nan 的问题,次要是输入的 box 大小小于指定的尺寸时呈现。(应用 dump 算子将该算子的输入输出都保留下来,独自运行该算子不会报错)
解决办法:
将该算子的默认参数都补上去,例如如下:
此时再运行,nan 的问题不复见。