当使用`keras.models.load_model()`加载模型时,Keras 2.0抛出“未知层:DenseFeatures”。

时间:2019-07-14 19:11:34

标签: tensorflow keras tensorflow2.0

我正在使用Tensorflow 2.0,构建一个包含DenseFeatures层的模型,并使用model.save()保存它。当我尝试用tf.keras.models.load_model()加载它时,出现以下错误:

ValueError: Unknown layer: DenseFeatures

如果模型不包含该层,则不会发生这种情况。

简短的Colab演示了成功的流程(具有输入和密集层的模型)和失败的流程(以DenseFeatures开头的模型):https://colab.research.google.com/drive/1ayHvyCaszw3RJenAeqVPwNBp1Y-c7Umj

有什么主意吗?

谢谢!

扎克

0 个答案:

没有答案