Export to ONNX Error

Hi, thanks for the great work. Following my question here , I tried to convert to ONNX using this repo. But I got several errors.

By inputting this command just like your example, I got segmentation fault error :

sudo python3 onnx_export.py --model mobilenetv3_100 ./mobilenetv3_100.onnx
==> Creating PyTorch mobilenetv3_100 model
==> Exporting model to ONNX format at './mobilenetv3_100.onnx'
==> Loading and checking exported model from './mobilenetv3_100.onnx'
Segmentation fault

When I tried with efficientnet_b0 using checkpoint and not using checkpoint sudo python3 onnx_export.py --model efficientnet_b0 ./efficientnet.onnx or sudo python3 onnx_export.py --model efficientnet_b0 --checkpoint ../train/model_best.pth.tar --num-classes 30 ./efficientnet.onnx

I got Couldn't export Python operator SwishAutoFn error.

==> Creating PyTorch efficientnet_b0 model
=> Loading checkpoint '../train/20191015-Deepeye36k-efficientnet_b0-224/model_best.pth.tar'
=> Loaded checkpoint '../train/20191015-Deepeye36k-efficientnet_b0-224/model_best.pth.tar'
==> Exporting model to ONNX format at './efficientnet.onnx'
Traceback (most recent call last):
  File "onnx_export.py", line 75, in <module>
    main()
  File "onnx_export.py", line 59, in main
    input_names=input_names, output_names=output_names)
  File "/home/ivan/.local/lib/python3.6/site-packages/torch/onnx/__init__.py", line 26, in _export
    result = utils._export(*args, **kwargs)
  File "/home/ivan/.local/lib/python3.6/site-packages/torch/onnx/utils.py", line 394, in _export
    operator_export_type, strip_doc_string, val_keep_init_as_ip)
RuntimeError: ONNX export failed: Couldn't export Python operator SwishAutoFn

Any help would be appreciated, thanks

[rwightman]

@ivder Good to hear. You should be able to use the models with any ONNX runtime or conversion tool (https://github.com/microsoft/onnxruntime, https://github.com/onnx/onnx-tensorrt, etc) that supports the same file format and operator versions as you export in. Caffe2 is just the default available runtime if you have PyTorch installed.

[rwightman]

@ivder It looks like the opposite of what I expected, PyTorch/ONNX being too new. I get the same segfault with PyTorch 1.3 + ONNX 1.6 installed. I get a different crash with PyTorch 1.3 + ONNX 1.5 installed. It works with PyTorch 1.2 + ONNX 1.5

Some related issues: https://github.com/onnx/onnx/issues/2417 https://github.com/onnx/onnx/issues/2394