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

Adding Raw ONNX support to SurrealML #48

Closed
wants to merge 25 commits into from
Closed

Adding Raw ONNX support to SurrealML #48

wants to merge 25 commits into from

Conversation

maxwellflitton
Copy link
Contributor

This pull request enables users to directly import raw ONNX models into SurrealML and package them as surml files using the following code:

file = SurMlFile(model=raw_onnx_model, name="linear", inputs=HOUSE_LINEAR["inputs"], engine=Engine.ONNX)

maxwellflitton and others added 25 commits March 13, 2024 10:04
* adding build using binary downloads (#8)

* adding build using binary downloads

* sorting out the build.rs

* updating build.rs for surrealml package

* prepping version for release

* now has target tracking (#10)

* adding check in build.rs for docs.rs

* removing build.rs for main surrealml to ensure that libraries using the core do not need to do anything in their build.rs

* adding machine learning pipelines for bioengineering projects at Kings College London

* Remove integrated_training_runner/run_env/ from tracking

* adding machine learning pipelines for bioengineering projects at Kings College London

* Update FFmpeg data access module and README (#29)

* adding run_env to the gitignore

---------

Co-authored-by: Yang Li <[email protected]>
* Develop (#35)

* adding glue module to enable error handling (#27)

* added batching runner (#28)

* Kings college london integration (#30)

* adding build using binary downloads (#8)

* adding build using binary downloads

* sorting out the build.rs

* updating build.rs for surrealml package

* prepping version for release

* now has target tracking (#10)

* adding check in build.rs for docs.rs

* removing build.rs for main surrealml to ensure that libraries using the core do not need to do anything in their build.rs

* adding machine learning pipelines for bioengineering projects at Kings College London

* Remove integrated_training_runner/run_env/ from tracking

* adding machine learning pipelines for bioengineering projects at Kings College London

* Update FFmpeg data access module and README (#29)

* adding run_env to the gitignore

---------

Co-authored-by: Yang Li <[email protected]>

* bumping the version

* updating the README and module

* updating the surrealml-core deployment workflow

* updating the surrealml-core deployment workflow

* updating cargo

---------

Co-authored-by: Sam Hillman <[email protected]>
Co-authored-by: Yang Li <[email protected]>

* merging error modules into the core

* merging error modules into the core

* merging error modules into the core

---------

Co-authored-by: Sam Hillman <[email protected]>
Co-authored-by: Yang Li <[email protected]>
* adding buffer out of index check

* adding buffer out of index check
* updating the naming and increasing tests around the meta data of the stored ML models

* updating the naming and increasing tests around the meta data of the stored ML models

* updating the naming and increasing tests around the meta data of the stored ML models
* stashing for branch switch

* adding tests for tensorflow

* adding tests for tensorflow

* fixing requirement conflicts

* fixing requirement conflicts

* fixing requirement conflicts

* fixing requirement conflicts

* fixing requirement conflicts
adding support for raw ONNX models
@maxwellflitton
Copy link
Contributor Author

This pull request enables users to directly import raw ONNX models into SurrealML and package them as surml files using the following code:

file = SurMlFile(model=raw_onnx_model, name="linear", inputs=HOUSE_LINEAR["inputs"], engine=Engine.ONNX)

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

Successfully merging this pull request may close these issues.

2 participants