Problem Converting Tensorflow Model To Tensorflow-lite (.tflite) Format

I made a tensorflow model in python for image classification. Im using Windows 10. I have a Train.py class where i define the graph in build_graph() and train the model in train().

Solution 1:

TOCO on Windows is problematic. I have faced such problems till I found a solution. The solution is to upload all the saved models or graphdef to a Google Colab notebook. Then,

1. Connect with a GPU or TPU runtime. ( Change Runtime Type option )
2. Upload the saved_model in the runtime.( Files section in the top-left corner )
3. Write the same script in one cell which you have mentioned.
4. Make sure to create the necessary directories in the runtime. See this answer.
5. The conversion will take place in the cloud.

So, no problems of TOCO. See this notebook for info.