RuntimeError: ONNX export failed: Couldn't export Python operator SwishImplementation

[nutsam]

I ran your convert to onnx code and had error like title, can you help?

[murdockhou]

@nutsam you need set EfficientSwish false like that

[nutsam]

Yes, I used your code and has that line already,
but still error.

[murdockhou]

Are you sure you are using the original code and not change anywhere? It's weird if you not change code but occur this error. The env that I use is: - ubuntu 16.04 - cuda 10.1 - cudnn 7.6 - pytorch 1.4 nutsam <[email protected]> 于2020年5月14日周四 上午10:08写道：

…

Yes, I used your code and has that line already, but still error. — You are receiving this because you commented. Reply to this email directly, view it on GitHub <#6 (comment)>, or unsubscribe <https://github.com/notifications/unsubscribe-auth/AEMCC7JUKIWDTRYUJF2FX5DRRNHCJANCNFSM4M7RQM7Q> .

[Beam-wi]

@nutsam Have you solved this problem? I encounter the same problem with you.

[Beam-wi]

@nutsam 知道了，作者在 EfficientDetBackbone 中实例化 BiFPN，Regressor，Classifier时没有传onnx_export参数，加进去就行了

[swchew]

@Beam-wi may I ask which code and exactly where should we insert "onnx_export" into?

[heitorrapela]

If you print the model (print(model)), you can debug in which parts of the code still have the MemoryEfficientSwish.

In my case, my model had the backbone (efficient net) and the detector wrapped, so when I loaded the set_swish after the loaded model or if I used the flag to initialize the model and then load the weights, flag onnx_export didn't update the layer. So, I had to fix it manually as the following code: