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

Suitable for deployment? #385

Open
tamis-laan opened this issue Mar 25, 2024 · 0 comments
Open

Suitable for deployment? #385

tamis-laan opened this issue Mar 25, 2024 · 0 comments

Comments

@tamis-laan
Copy link

I'm looking to deploy torch models in production and it would be amazing if we could use go for this. As you know containerizing python and pytorch creates a lot of overhead. Not only in terms of memory usage but also the container image is more than a gigabyte in size.

I was looking at https://github.com/orktes/go-torch using torch script but noticed it's out of date and refers to this library. However this library looks more like a complete pytorch replica then for simply deploying models trained in pytorch.

So I'm wondering is this library suitable for deploying torch models? And if so is there some example code? And what fileformat should we use. I believe that pytorch is using pickle to save models, but I'm pretty sure that's not going to work with this go library? There is also torch script but I can't find any reference to it. Or maybe 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

No branches or pull requests

1 participant