RuntimeError: step!=1 is currently not supported
原因主要是低版本的opset不支持切片操作导致的;
把模型转换的代码改成如下所示即可,即使用版本11以上的opset:
torch.onnx.export(model, img, "xxx.onnx", verbose=True,opset_version=11,export_params=True)
2. 解决该问题后可能会继续出现如下错误:
RuntimeError: Exporting the operator silu to ONNX opset version 11 is not supported. Please open a bug to request ONNX export support for the missing operator.
这是因为onnx与pytorch一些方法不兼容导致的,onnx不支持silu,把激活函数换一种写法即可。
解决方法如下:
pytorch的激活函数源码位置:X:\anaconda3\envs\python3.7\Lib\sitepackages\torch\nn\modules\activation.py(此处结合自己的anaconda实际安装位置来更改)。
修改成如下:
class SiLU(Module):
__constants__ = ['inplace']
inplace: bool
def __init__(self, inplace: bool = False):
super(SiLU, self).__init__()
self.inplace = inplace
def forward(self, input: Tensor) -> Tensor:
# ------------------------------------- #
# 把F.silu替换掉,修改后如下
return input * torch.sigmoid(input)
#原来的代码
return F.silu(input, inplace=self.inplace)
————————————————
原文链接:https://blog.csdn.net/weixin_37889356/article/details/121671974
版权说明 : 本文为转载文章, 版权归原作者所有 版权申明
原文链接 : https://blog.csdn.net/jacke121/article/details/125090226
内容来源于网络,如有侵权,请联系作者删除!