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

Feat: save model as onnx option #137

Merged
merged 4 commits into from
Aug 20, 2024
Merged

Feat: save model as onnx option #137

merged 4 commits into from
Aug 20, 2024

Conversation

ankeko
Copy link
Collaborator

@ankeko ankeko commented Aug 19, 2024

📥 Pull Request Description

Additionally to saving the model as tf files, there is now an option to save them in the general ONNX format.

👀 Affected Areas

Training, Model Callbacks

📝 Checklist

Please make sure you've completed the following tasks before submitting this pull request:

  • Pre-commit hooks were executed
  • Changes have been reviewed by at least one other developer
  • Tests have been added or updated to cover the changes (only necessary if the changes affect the executable code)
  • All tests ran successfully
  • All merge conflicts are resolved
  • Documentation has been updated to reflect the changes
  • Any necessary migrations have been run

@ankeko ankeko changed the base branch from main to develop August 20, 2024 07:06
@ankeko ankeko added the ✨ feature Requests or discussions about new features label Aug 20, 2024
pyproject.toml Outdated Show resolved Hide resolved
@ankeko ankeko enabled auto-merge (rebase) August 20, 2024 12:55
@ankeko ankeko merged commit e0becd7 into develop Aug 20, 2024
6 checks passed
@ankeko ankeko deleted the feature/save_onnx_option branch August 20, 2024 13:13
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
✨ feature Requests or discussions about new features
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants