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

Inference in go (trained in python and deploy in golang) #529

Open
manjunathsc24 opened this issue Jul 21, 2022 · 1 comment
Open

Inference in go (trained in python and deploy in golang) #529

manjunathsc24 opened this issue Jul 21, 2022 · 1 comment

Comments

@manjunathsc24
Copy link

Hello @chewxy ,

I have a model(autoencoder) trained in python , now i would like to deploy (do inference) in golang, i tried with onnx-go but that is not good in performance , could you please suggest me how to do inference in golang gorgonia that will be great help from your side.

Thank You
Manjunath

@auxten
Copy link
Contributor

auxten commented Nov 3, 2022

I think it's pretty difficult(impossible) to import a model trained by other frameworks to gorgonia.
For now, gorgonia build the computing graph arch in golang, the Forward and Backprop behavior may not identical.
To achieve that, there are at least 3 things to do manually:

  1. Parse the model description for PyTorch or TF
  2. Build the missing Node type needed by autoencoder
  3. Create the whole computing graph and recover the weights, maybe something like https://github.com/gorgonia/parser/

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

2 participants