使用以下代码:
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())
解决成功: