Tensorflow tf.keras.models.load_model() 打开h5文件失败
Posted ʚVVcatɞ
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了Tensorflow tf.keras.models.load_model() 打开h5文件失败相关的知识,希望对你有一定的参考价值。
使用以下代码:
import tensorflow as tf
new_model = tf.keras.models.load_model('./less_model.h5')
print(new_model.summary())
此方法保存以下所有内容:
- 1.权重值
- 2.模型配置(架构)
- 3.优化器配置
出现问题:
解决方法:
import tensorflow as tf
import h5py
file = h5py.File('less_model.h5')
new_model = keras.models.load_model(file)
print(new_model.summary())
解决成功:
以上是关于Tensorflow tf.keras.models.load_model() 打开h5文件失败的主要内容,如果未能解决你的问题,请参考以下文章
HCIA-AI_深度学习_TensorFlow2模块tf.keras基本用法
HCIA-AI_深度学习_TensorFlow2模块tf.keras基本用法
tf.keras model.fit():在相同数据上火车损失和val损失之间的巨大差异