pytorch转onnx: step = 1 is currently not supported以及Exporting the operator silu to ONNX opset version

1. 在yolov5s的pytorch模型转换onnx模型时报如下错误:

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)

  • 19
    点赞
  • 22
    收藏
    觉得还不错? 一键收藏
  • 8
    评论
评论 8
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值