android - 如何修复此模块 'tensorflow' 没有属性 'lite'？

Question

我从https://www.tensorflow.org/lite/convert/python_api复制了代码在运行代码时，它给了我这个错误模块“tensorflow”没有属性“lite”

我还没有看到任何可以解决我的问题的教程，所以我想在这里问问是否有人可以帮助我解决我的问题

这是从 tensorflow lite 官方文档中复制的代码，当我尝试运行它时，它给我这个错误 "module 'tensorflow' has no attribute 'lite'"

import tensorflow as tf

img = tf.placeholder(name="img", dtype=tf.float32, shape=(1, 64, 64, 3))
var = tf.get_variable("weights", dtype=tf.float32, shape=(1, 64, 64, 3))
val = img + var
out = tf.identity(val, name="out")

with tf.Session() as sess:
  sess.run(tf.global_variables_initializer())
  converter = tf.lite.TFLiteConverter.from_session(sess, [img], [out])
  tflite_model = converter.convert()
  open("converted_model.tflite", "wb").write(tflite_model)

Answer 1

很抱歉给您带来不便，您可以通过下载 nightly 来修复它。

编辑:或使用 tf.contrib.lite 也应该有效。