Cannot convert model to tflite about tensorflow

[Venkat6871]

Hi @mdomaradzki-cosmose ,
Sorry for the delay, @LakshmiKalaKadali the issue Was able to reproduce the issue on Colab using TF v2.15 , 2.16 , and TF-nightly ,Please find the gist here for reference.

Thank you!

[mdomaradzki-cosmose]

Thanks for your response

[LakshmiKalaKadali]

Hi @mdomaradzki-cosmose,

The error is due to Keras 3. As a workaround install Keras 2 as follows.

    pip install -U tf_keras # Keras 2
    import os
    os.environ["TF_USE_LEGACY_KERAS"] = "1" 

Here is the gist with TF 2.16 and keras2 workaround. The TFLite model is successfully converted.

Thank You

[mdomaradzki-cosmose]

Thank you, It works!

[LakshmiKalaKadali]

Hi @mdomaradzki-cosmose,

Glad to know, the issue is resolved. Please feel free to close the issue.

Thank You