EfficientNet-PyTorch icon indicating copy to clipboard operation
EfficientNet-PyTorch copied to clipboard

Published 20 hours ago • verified publisher icon lukemelas

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

Open DJMeng opened this issue 5 years ago • 2 comments

When I exported b7 to onnx: my env is torch 1.5, opset11.and I try other's version of torch or opset, it will meet other problems. someone meet this error? Please help. Thanks.

DJMeng avatar Jun 30 '20 09:06 DJMeng

have you solved this issue? The collab notebook doesn't work either because of Swish not exported to ONNX

mikeseven avatar Sep 23 '20 23:09 mikeseven

found it, solution is here: model.set_swish(memory_efficient=False) see https://github.com/lukemelas/EfficientNet-PyTorch/issues/91

mikeseven avatar Sep 23 '20 23:09 mikeseven