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

exporting to torch script #1873

Open
mohamedsaeed8223 opened this issue Feb 23, 2025 · 1 comment
Open

exporting to torch script #1873

mohamedsaeed8223 opened this issue Feb 23, 2025 · 1 comment
Labels
framework: pytorch Related to PyTorch backend type: enhancement Improvement type: new feature New feature

Comments

@mohamedsaeed8223
Copy link

mohamedsaeed8223 commented Feb 23, 2025

🚀 Adding torchscript export support

are there any plans to support exporting models using torch script?

Motivation, pitch

In my experience I always felt like the loss of performance in onnx export is much higher than it is in torch script exports
While both very similar in inference speed

@felixdittrich92 felixdittrich92 added framework: pytorch Related to PyTorch backend type: new feature New feature labels Feb 24, 2025
@mindee mindee deleted a comment from felixT2K Feb 24, 2025
@felixdittrich92
Copy link
Contributor

Hi @mohamedsaeed8223 👋,

Currently it's not planned to add torchscript support, because I think we will run into the same issues we have had with Onnx - that's why OnnxTR exists - (postprocessing isn't traceable / exportable - because of several opencv operations)

But if you would like to work on that feel free to give it a try - with a PR I can help with the integration 👍

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
framework: pytorch Related to PyTorch backend type: enhancement Improvement type: new feature New feature
Projects
None yet
Development

No branches or pull requests

2 participants