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.predict 导致内存泄漏

tf.keras model.fit():在相同数据上火车损失和val损失之间的巨大差异

《30天吃掉那只 TensorFlow2.0》4-5 AutoGraph和tf.Module

《30天吃掉那只 TensorFlow2.0》4-5 AutoGraph和tf.Module