-
Notifications
You must be signed in to change notification settings - Fork 3k
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
Add support for Less in GNMT #1700
Comments
I am using the latest master from today: commit f25847b |
This is for float64 input, I will create a MR for this. |
@snnn How do I create a MR? I don't seem to have permission |
How is your implementation different from this PR? #1604 |
this model have double (float 64) |
We found the missing input type at this line: |
Hi @RanACohen,
|
@neginraoof In the GNMT I posted in the drive link above (made by NVidia I think) there is a casting to float64 (double) and then Less. |
Where did you get the model? How did you convert it? Is it the same one in mlperf https://github.com/mlperf/inference ? Welcome to submit your change. I can help review. |
@RanACohen The onnx model you are trying to run in the Google Drive is created by me. This is the onnx model exported from a fork of GNMT model as a workaround. Now that 'int64' is supported for 'Less' op, you should be able to run the model with no issues. I exported the onnx model from GNMT master branch and uploaded it again (replaces the files). You can try running the new one. There's no need to use double for 'Less' op in this model. Sorry if the uploaded onnx model confused you. |
@neginraoof Thanks a lot! Does not hurt to extend the runtime with Less(double) in any case 😃 |
When I use GNMT from https://drive.google.com/drive/folders/1cWYUg1V1FFgFo8dF-W0hR2tyezCzY_UV
I get:
runtimeError: [ONNXRuntimeError] : 9 : NOT_IMPLEMENTED : Could not find an implementation for the node Less(9)
The text was updated successfully, but these errors were encountered: