解决tensorflow:No training configuration found in the save file, so the model was not compiled

最新推荐文章于 2023-06-05 20:53:42 发布

鹏阿鹏

最新推荐文章于 2023-06-05 20:53:42 发布

阅读量3.1k

点赞数 4

分类专栏：机器学习与深度学习 TensorFlow2 文章标签： tensorflow 深度学习 python

本文链接：https://blog.csdn.net/AwesomeP/article/details/126831850

版权

18 篇文章 1 订阅

订阅专栏

16 篇文章 4 订阅

订阅专栏

TensorFlow2：通过加载文件（keras.models.load_model）使用已经训练好的模型时出现如下警告

WARNING:tensorflow:No training configuration found in the save file, so the model was not compiled. Compile it manually.

猜测出现该问题的原因：我们在原本训练模型时使用了自定义的训练方法，没有直接使用complie、fit函数来训练模型，导致保存模型时出现No training configuration found in the save file

解决方法：load_model时添加参数compile=False

model_file_path = './SaveModel/tf2keras_xxxx_model.h5'
model = keras.models.load_model(model_file_path,compile=False)

成功解决

优惠劵

关注关注