gpt4 book ai didi

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

转载 作者:行者123 更新时间:2023-12-04 13:15:55 26 4
gpt4 key购买 nike

我已经创建了自己的类子类 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/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com