-
Notifications
You must be signed in to change notification settings - Fork 12
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
Rave torch to onnx #2
Comments
Hi @nagra00, |
So this means that it is not possible to convert an already trained torch model to onnx? Static or dynamic onnx makes no difference to me. For MacOS 10.13 i have to compile Scyclone anyway. I took a quick look at the dev branch and noticed that you have already adapted the code to the onnx API change. |
Yes it is possible that models which have been trained without these flags can't be converted to onnx properly. But more info on that very soon.
Yes, thanks for the hint! |
Hi @nagra00! As already mentioned above specific flags are required to be passed to the training script in order to successfully export onnx models for Scyclone. It is still probable to export already trained RAVE models to onnx without those flags. I assume you have already tried: rave export_onnx --run /path/to/the/run/directory If the script returns an error, could you please send over the training .gin file as well as the error message so we can take a look at it? Thanks for your patience in the meantime! |
Hi, I have not trained any models myself, it just takes way too long. What I was looking for is a conversion of the existing RAVE models to onnx. I thought this would be possible by keeping the tensors and replacing the algebraic parts with appropriate onnx versions. But as I understand it, this is not possible. |
There are some existing onnx RAVE models you can use with Scyclone: |
Congratulations on this really great application.
I am using the latest version of onnx and had to make a small adjustment due to an API change.... no big deal.
It would be nice if you could put a script to convert RAVE torch models to onnx in the repository.
The text was updated successfully, but these errors were encountered: