我已经创建了自己的类子类 tf.keras.Model并且正在尝试保存和加载它的训练实例。我正在尝试关注 this教程但每次我去加载保存的模型时，我都会收到相同的错误消息:TypeError: __init__() got an unexpected keyword argument 'reduction' .我已经尝试将该关键字参数添加到我的类中，但它没有任何改变。有任何想法吗？
最佳答案
我在 Tensorflow 1.14 上遇到了同样的问题，并通过添加 compile=False 解决了它到加载功能:
new_model = tf.keras.models.load_model('saved_model/my_model', compile=False)
即使与
compile=False可以运行 model.predict()功能。解决方案来源于this Tensorflow issue .
关于python - 使用 Tensorflow 2.1 加载自定义模型，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/60530304/