How to tranform efficient-pytorch to efficient-onnx

[peyer]

I have tried to convert efficient-pytorch to efficient-onnx with api (torch.onnx.export), but I meet a problem showing below info
Failed to export an ONNX attribute, since it's not constant, please try to make things (e.g., kernel size) static if possible
How could I fix it ?

[lukemelas]

Yes, I can try to make the kernel size static in the next significant update.

[peyer]

@lukemelas Thansk!

[lukemelas]

Done! I had to change some things with the padding, so it took a little while.

Here is an example:
https://colab.research.google.com/drive/1rOAEXeXHaA8uo3aG2YcFDHItlRJMV0VP

[lukemelas]

Closing this issue, but definitely re-open if you have any issues.

Also, let me know if exporting to ONNX / mobile works for you!

[peyer]

@lukemelas Successd and thanks!

[lukemelas]

Great to hear!

[deepconsc]

@lukemelas Hey! Appreciate your great work.
For the latest version, exporting to onnx fails due to Swish. Any ideas?

[bkhti4]

Please use model.set_swish(memory_efficient=False).

[kendyChina]

@bkhti4 Thanks! When I replaced SwishImplementation() with Swish(), I converted the model to onnx successfully.

[lukemelas]

How there's nn.SiLU as well, so I believe all models should be memory-efficient and exportable :)

The older functions are still included for backward compatibility with old versions of PyTorch

[fangliang425]

@lukemelas Hi, when I simply ran

model = EfficientNet.from_name(model_name='efficientnet-b0')
model.eval()
model.set_swish(memory_efficient=False)
torch.onnx.export(model, torch.rand(10, 3, 240, 240), "EfficientNet-B0.onnx")

, and got error as blow, any ideas? Thanks.
RuntimeError: Exporting the operator silu to ONNX opset version 9 is not supported. Please open a bug to request ONNX export support for the missing operator.

environment:
Ubuntu 20.04
torch 1.7.1
cuda 11.0
efficientnet-pytorch 0.7.1

[aojue1109]

@lukemelas嗨，当我简单地跑步时

model = EfficientNet.from_name(model_name='efficientnet-b0')
model.eval()
model.set_swish(memory_efficient=False)
torch.onnx.export(model, torch.rand(10, 3, 240, 240), "EfficientNet-B0.onnx")

，并因错误而遭受打击，有什么主意吗？谢谢。
RuntimeError: Exporting the operator silu to ONNX opset version 9 is not supported. Please open a bug to request ONNX export support for the missing operator.

环境：
Ubuntu 20.04
火炬1.7.1
cuda 11.0
高效net-pytorch 0.7.1

Is it resolved? I need your help！

[chilin0525]

@aojue1109 I converted the model to onnx successfully. There's my gist link.
If still not solve your problem I think you should re-open the issue.