es.davy.ai

Preguntas y respuestas de programación confiables

¿Tienes una pregunta?

Si tienes alguna pregunta, puedes hacerla a continuación o ingresar lo que estás buscando.

El módulo ‘tensorflow._api.v2.lite’ no tiene el atributo ‘TFliteConverter’.

Estoy intentando convertir un modelo de TensorFlow a un modelo de TensorFlow Lite.

converter = tf.lite.TFliteConverter.from_keras_model(model)
converter.optimizations = [tf.lite.Optimize.OPTIMIZE_FOR_SIZE]
tflite_model = converter.convert()
open(tflite_model_name + '.tflite', 'wb').write(tflite_model)

Pero obtengo el error “el módulo ‘tensorflow._api.v2.lite’ no tiene el atributo ‘TFliteConverter'”. ¿Es porque uso LSTM en mi modelo?

Tags:  ,

Answer

  1. Avatar for davy.ai

    Es poco probable que el error sea causado por el uso de LSTM en tu modelo. El mensaje de error sugiere que el módulo tensorflow._api.v2.lite no tiene un atributo llamado TFliteConverter. Esto podría ser debido a una discrepancia de versión o a una instalación de paquetes faltante. Para solucionarlo, puedes intentar actualizar tensorflow y tensorflow-lite a la última versión o reinstalar los paquetes. Otra solución es importar TFLiteConverter desde tensorflow.lite en lugar de tensorflow._api.v2.lite, como se muestra a continuación:

    from tensorflow.lite import TFLiteConverter
    
    converter = TFLiteConverter.from_keras_model(model)
    converter.optimizations = [tf.lite.Optimize.OPTIMIZE_FOR_SIZE]
    tflite_model = converter.convert()
    open(tflite_model_name + '.tflite', 'wb').write(tflite_model)
    

    Esto debería resolver el error y permitirte convertir tu modelo de tensorflow a tensorflowlite.

Comments are closed.