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

Freezing model possibility #30

Closed
arskama opened this issue Mar 11, 2019 · 2 comments
Closed

Freezing model possibility #30

arskama opened this issue Mar 11, 2019 · 2 comments

Comments

@arskama
Copy link

arskama commented Mar 11, 2019

Hi,
Thanks first for the nice work you made!

feature request:
I was planning to try the result of your work on tensorflow.js
For this I need to convert the model from tensorflow to tensorflow.js, which means that I need to freeze the model before that.
Apparently, from what I found out, freezing the model means that I need to have at least the following data:
--input_graph_def filename.pbtxt
--input_checkpoint filename.ckpt
--output_node_names
--output_graph filename.pb

see:
https://github.com/tensorflow/tensorflow/blob/master/tensorflow/python/tools/freeze_graph.py

At the moment, there is only the --input_checkpoint filename.ckpt available.

Describe the solution you'd like
Is there a way you could upload the filename.pbtxt and output_node_names?

What do you think?
Thanks again!
Arno

@arskama
Copy link
Author

arskama commented Mar 11, 2019

For whatever reason, I just skipped the fact that you are using PyTorch!!!

@avinashpaliwal
Copy link
Owner

For whatever reason, I just skipped the fact that you are using PyTorch!!!

Yes. So you will have to use some PyTorch to Tensorflow converter : )

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