TensorFlowASR
TensorFlowASR copied to clipboard
RNN-Transducer TFlite conversion Broken
Hello,
Im trying to convert a pertained model trained on librispeech, but I run into the following error
tensorflow.lite.python.convert_phase.ConverterError: input resource[0] expected type resource != float, the type of streaming_transducer_greedy_while_streaming_transducer_decoder_streaming_transducer_prediction_embedding_embedding_lookup_11394_0[0]
In {{node streaming_transducer_greedy/while/streaming_transducer_decoder/streaming_transducer_prediction_embedding/embedding_lookup}}
Failed to functionalize Control Flow V1 ops. Consider using Control Flow V2 ops instead. See https://www.tensorflow.org/api_docs/python/tf/compat/v1/enable_control_flow_v2.
to run the tflite conversion, I ran the below code
python3 examples/rnn_transducer/tflite.py --saved Librispeech/25.h5 --subwords output rnnt.tflite
Platform OS : MacOS 10.15.7 Python : 3.9 Tensorflow-cpu : 2.6.3
Could you suggest a possible way to convert a model to tflite as installing tf-nightly causes dependency issues and does not allow any code to be run at all.
Thank You!
Any updated solution here? I encountered the same issue here with tf2.9.
This bug has been resolved https://github.com/tensorflow/tensorflow/issues/42410#issuecomment-1201150119. Basically, we just need to load model while are are loading the concrete function, i.e., change this line to converter = tf.lite.TFLiteConverter.from_concrete_functions([concrete_func], model).