Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

export to onnx #30

Open
carr123 opened this issue Apr 29, 2021 · 1 comment
Open

export to onnx #30

carr123 opened this issue Apr 29, 2021 · 1 comment

Comments

@carr123
Copy link

carr123 commented Apr 29, 2021

I build a neural network using ScatLayer as one layer.
when I export the model from pytorch to onnx format, error occurs.

pytorch_wavelets\utils.py", line 162, in reflect
out = np.where(normed_mod >= rng, rng_by_2 - normed_mod, normed_mod) + minx
TypeError: '>=' not supported between instances of 'numpy.ndarray' and 'Tensor'

@fbcotter
Copy link
Owner

fbcotter commented Aug 6, 2021

Ah bummer. Yeah I remember I had to do some funky things to get padding to work in the right way. It's several months later I know, but what command did you use to export to onnx? It might be possible to do the padding with torch tensors so the export works fine.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants