python - 使用 TensorFlow 2.1 加载自定义模型

Question

我已经创建了自己的类子类tf.keras.Model，并正在尝试保存和加载经过训练的实例。我正在尝试按照本教程进行操作，但每次加载保存的模型时，我都会收到相同的错误消息：TypeError: __init__() got an unexpected keyword argument 'reduction'. 我已经尝试将该关键字参数添加到我的课程中，但它没有任何改变。有任何想法吗？

标签： pythontensorflowsave

Answer 1

我对Tensorflow 1.14有同样的问题，并通过添加compile=False到加载函数来解决它：

new_model = tf.keras.models.load_model('saved_model/my_model', compile=False)

即使compile=False可以运行该model.predict()功能。

解决方案源于这个 Tensorflow 问题。