Skip to content

RNN-Transducer TFlite conversion Broken  #247

Open
@SupreethRao99

Description

@SupreethRao99

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!

Metadata

Metadata

Assignees

No one assigned

    Labels

    bugSomething isn't workingneed to reproduceNeed a code or time to reproduce the issue

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions