search

fbcotter / pytorch_wavelets

Pytorch implementation of 2D Discrete Wavelet (DWT) and Dual Tree Complex Wavelet Transforms (DTCWT) and a DTCWT based ScatterNet
Other
943 stars 146 forks source link

Export DWTForward to onnx #38

Closed hwenjun18 closed 2 years ago

hwenjun18 commented 2 years ago

HI, nice work!

I try to export a pytorch model which contains DWTForward module, just as logs shows, an error occurs, after go deep into source code, I find that [AFB2D] is a python function which is consit of a set of pytorch op, which can not be exported to onnx? Do you have any idea how to export this [AFB2D] to onnx model? thank you!

> RuntimeError: ONNX export failed: Couldn't export Python operator AFB2D

kafan1986 commented 1 year ago

@hwenjun18 Where you able to export the model to onnx?