使用 Tensorflow 2.1 加载自定义模型

Question

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

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 问题。