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

Doing inference/prediction only, with weights loaded from a Tensorflow NN #22

Open
josephernest opened this issue Jun 22, 2021 · 1 comment

Comments

@josephernest
Copy link

josephernest commented Jun 22, 2021

Nice project!
Is it possible to do the following:

  1. Train a CNN with, for example, Tensorflow + Python
  2. Save the model weights into a file
  3. Load this weights file from MiniDNN / C++ code, and do prediction/inference only with MiniDNN (no training done with MiniDNN)

How would it be possible to load weights/coefficients of a model pre-trained with Tensorflow, into MiniDNN?

@TimJones7
Copy link

I'd also love to know this answer.

I haven't yet implemented it, but I've also found cppflow , which is a lightweight wrapper for tensorflow, if you're wanting to load a trained model into c++ code to draw inference from.

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