Python 使用Tensorflow 2.1加载自定义模型
我已经创建了自己的类子类Python 使用Tensorflow 2.1加载自定义模型,python,tensorflow,save,Python,Tensorflow,Save,我已经创建了自己的类子类tf.keras.Model,并试图保存和加载一个经过训练的实例。我试图按照教程进行操作,但每次加载保存的模型时,都会收到相同的错误消息:TypeError:\uuu init\uuuu()得到一个意外的关键字参数“reduce”。我已经尝试将那个关键字参数添加到我的类中,但它没有任何改变。有什么想法吗?我对Tensorflow 1.14也有同样的问题,通过向加载函数添加compile=False解决了这个问题: new_model=tf.keras.models.loa
tf.keras.Model
,并试图保存和加载一个经过训练的实例。我试图按照教程进行操作,但每次加载保存的模型时,都会收到相同的错误消息:TypeError:\uuu init\uuuu()得到一个意外的关键字参数“reduce”
。我已经尝试将那个关键字参数添加到我的类中,但它没有任何改变。有什么想法吗?我对Tensorflow 1.14也有同样的问题,通过向加载函数添加compile=False
解决了这个问题:
new_model=tf.keras.models.load_model('saved_model/my_model',compile=False)
即使使用compile=False
也可以运行model.predict()
函数
解决方案源自于