TensorFlow Lite:初始化节点不存在

时间:2019-01-10 04:40:29

标签: tensorflow

我试图使用以下代码将Keras文件(.h5)中的模型转换为TensorFlow Lite文件(.tflite):

data = np.zeros((129, 326, 490, 3), dtype=np.uint8)

但是,以下行:

# Save model as .h5 keras file
keras_file = "eSleep.h5"
model_save = tf.keras.models.save_model(model,keras_file,overwrite=True,include_optimizer=True)

# Export keras file to TensorFlow Lite model
converter = tf.lite.TFLiteConverter.from_keras_model_file(keras_file)
tflite_model = converter.convert()
open("eSleep.tflite", "wb").write(tflite_model)

返回的错误:

tflite_model = converter.convert()

有人可以帮助我了解“图形中不存在初始化节点密集/内核/分配”的含义是什么以及如何解决该错误吗?

2 个答案:

答案 0 :(得分:0)

夜间使用时,我也遇到同样的问题。我通过使用tensorflow 1.12解决了它。

答案 1 :(得分:0)

以我的经验,即使显示此错误,转换后的模型也可以正常工作。您可以忽略该错误。