-
Couldn't load subscription status.
- Fork 9
Produce ML prediction of virtual photon four momentum, implemented in ONNX #278
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
base: main
Are you sure you want to change the base?
Conversation
…nsor names (need to check if onnx has some better workaround, or if I can fix these for all models)
|
Libraries: Headers are in |
|
Looking into if that library works but not sure, I can't find much documentation on the difference between ONNX IFI and ONNX-runtime, which is what I had been using. |
If the ONNX version in If you need the newer version with |
|
There are two separate packages here: onnx and onnxruntime. Because we build everything in spack, we use what is in spack. Onnx (py-onnx) is up to date, but onnxruntime (py-onnxruntime) isn't. That needs some more of my efforts in spack/spack#37064. Also, onnxruntime is large and we probably don't necessarily want to include it unless someone is actively using it. |
|
Or I of course welcome any help in that spack recipe :-) |
Briefly, what does this PR introduce?
Previously implemented ML predictions for kinematics in pybind-11. This PR changes predictions to be carried out using ONNX (https://onnxruntime.ai).
Script added in
epic-analysis/depsto download and unpack onnx-runtime. Building with enabled withINCLUDE_ONNX=1 makeStill working to produce some example trained model to include as a part of a tutorial, but could be part of another PR.
What kind of change does this PR introduce?
Please check if this PR fulfills the following:
Does this PR introduce breaking changes? What changes might users need to make to their code?
Should not break by default. No changes should be necessary.
Does this PR change default behavior?
No.