pytorch转onnx时,pytorch中用了F.grid_sample,报错:
‘ONNX export failed on ATen operator grid_sampler because torch.onnx.symbolic.grid_sampler does not exist’

解决方案
pytorch的grid_sample操作是无法转换成onnx的算子的(暂时也没看到自写插件的)。

Logo

开放原子开发者工作坊旨在鼓励更多人参与开源活动,与志同道合的开发者们相互交流开发经验、分享开发心得、获取前沿技术趋势。工作坊有多种形式的开发者活动,如meetup、训练营等,主打技术交流,干货满满,真诚地邀请各位开发者共同参与!

更多推荐