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

[Feature request] Add kwargs or additional options for torch.onnx.export #2009

Open
martinkorelic opened this issue Sep 3, 2024 · 0 comments

Comments

@martinkorelic
Copy link

Feature request

In optimum.exporters.onnx.convert import export_pytorch, there could be an option to add additional kwargs to the function which could be passed to the torch.onnx.export function.

Motivation

If such an option possible or will this ruin any of the other features, or is there a reason why there is no option available as of yet?

Your contribution

Could contribute if this doesn't ruin any other features, or the current feature.

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

No branches or pull requests

1 participant