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的算子的(暂时也没看到自写插件的)。
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的算子的(暂时也没看到自写插件的)。
上一篇:libkmod: ERROR ../libkmod/libkmod-config.c:635 kmod_config_parse: /etc/modprobe.d/blacklist.conf li
下一篇:RuntimeError: ONNX export failed: Couldn‘t export operator aten::upsample_bilinear2d
pytorch,CUDA是否可用,查看显卡显存剩余容量
还是模型转换相关的呀,分享一些坑首先,简单说明一下pytorch转onnx的意义。在pytorch训练出一个深度学习模型后,需要在Tenso
举报文章
请选择举报类型
补充说明
0/200
上传截图
格式支持JPEG/PNG/JPG,图片不超过1.9M