我已经训练了一个模型并将其保存为.h5文件。由于我想在我的Android应用程序中使用它,我正在尝试在Colab上将其转换为tflite。这是我的代码:
我遇到的错误是:
AttributeError: type object 'TFLiteConverter' has no attribute 'from_keras_model'
如何解决这个问题?
回答:
您正在使用Colab的默认tensorflow
版本,即1.15.0
。
您可以通过在Colab的单元格中输入以下命令来下载最新版本:
!pip install --upgrade pip && pip install tensorflow
。
您可能需要先通过以下命令卸载tensorflow
版本1.15.0:
!pip uninstall tensorflow (或 tensorflow-gpu)
。
之后,您应该可以正常运行tf2.x
代码了。
注意:不久之后,TF2.0 将成为默认版本,此解决方法只是临时的。