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

Conversion to TF lite is failing #8

Open
sandeepjana opened this issue Nov 24, 2018 · 1 comment
Open

Conversion to TF lite is failing #8

sandeepjana opened this issue Nov 24, 2018 · 1 comment

Comments

@sandeepjana
Copy link

First, thanks for detailed description of training steps. I trained the model for a few epochs.

My final goal is to obtain tflite model. So, I converted the ckpt model to pb by following this SO answer:
https://stackoverflow.com/a/45868106/1595171

Then tried toco converter to convert the model to tflite, like this:

toco
--graph_def_file=logs/samples/output_graph_122303.pb
--output_file=output_graph_122303.tflite
--output_format=TFLITE
--input_shape=1,224,224,3
--input_array=input
--output_array='split:0', 'split:1', 'mul_60:0', 'mul_63:0', 'mul_61:0', 'mul_64:0', 'mul_62:0', 'mul_65:0'
--inference_type=FLOAT
--inference_input_type=FLOAT

I got the following error:
ValueError: Input 0 of node MobilenetV2/Conv/weights/Assign was passed float from MobilenetV2/Conv/weights:0 incompatible with expected float_ref.

I googled for quite sometime but nothing worked out for me.

@sydsim
Copy link
Owner

sydsim commented Dec 29, 2018

sorry but I can't help you because I don't know about the tf lite. but it seems that types float and float_ref doesn't match is the cause of the error.

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