【报错】ONNX: export failure ❌ 0.3s: Exporting the operator silu to ONNX opset version 11 is not support

原因

onnx不支持silu

解决

vscode进入

1. 进入conda/lib/python3.8/site-packages/torch/nn/modules/activation.py
2. 查找 class SILU()
3. 修改forward函数，return input * torch.sigmoid(input)

linux命令

因为我的ssh出问题了，vscode不能连接服务器，所以只好输入命令进入了

1. 进如/opt/conda/lib/python3.8/site-packages/torch/nn/modules目录

cd /opt/conda/lib/python3.8/site-packages/torch/nn/modules

2. vim 打开activation.py

(base) root@test:/opt/conda/lib/python3.8/site-packages/torch/nn/modules# vim activation.py 

3. 找到第 361 行 SILU 类， 如下图所示
   
4. 找到 forward函数，并按下 “i” 进入编辑模式
   
5. 修改为 return input * torch.sigmoid(input)
   
6. 按下 ESC 退出编辑模式，并依次按下":wq" 回车进行保存。如下图所示，左下角娴熟输入的“:wq”
   至此，修改完成，如下图onnx转换成功