admin管理员组

文章数量:1618706

项目场景：

提示：eror>>> ONNX: export failure: Exporting the operator silu to ONNX opset version 12 is not supported. Please open a bug to request ONNX export support for the missing operator.

问题描述

onnx不支持silu：

解决方案：

找到python路径/Lib/site-packages/torch/nn/modules/activation.py：

 重写394行
    def forward(self, input: Tensor) -> Tensor:
        # 原始
        # return F.silu(input, inplace=self.inplace)
        # 重写
        return input * torch.sigmoid(input)

本文标签： Exportingoperatorfailureonnxexport

版权声明：本文标题：ONNX: export failure: Exporting the operator silu to ONNX opset version 12 is not suppo 内容由热心网友自发贡献，该文观点仅代表作者本人， 转载请联系作者并注明出处：https://www.elefans.com/dongtai/1728785039a1173218.html， 本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容，一经查实，本站将立刻删除。