我正在使用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
有什么主意吗?
谢谢!
扎克