解决tf.keras.models.load_model加载模型报错问题
作者:_不负时光
这篇文章主要介绍了解决tf.keras.models.load_model加载模型报错问题,具有很好的参考价值,希望对大家有所帮助。如有错误或未考虑完全的地方,望不吝赐教
tf.keras.models.load_model 加载模型报错记录
Unknown activation function:leaky_relu
解决办法
是在tf.keras.models.load_model()括号里面加上
custom_objects = {“YYY”:XXX对应的函数或者对象}
例如报错
ValueError: Unknown activation function:leaky_relu
就加上
new_model=tf.keras.models.load_model("models/model.h5", custom_objects = {"leaky_relu":tf.nn.leaky_relu})
keras.models.load_model无法打开h5文件
是个小萌新,用tensorflow.keras.models.load_model打开h5文件时总是失败,各位大佬的解决方法过于深奥,实在看不懂,最后自己用下面办法解决的
1.可能是h5py与tensorflow版本不匹配,试试使用h5py2.10版本
2.版本调整后:
from tensorflow import keras model=keras.models.load_model('model.h5')
运行代码仍然出现以下问题:
更改代码如下:
from tensorflow import keras import h5py import os file=h5py.File('model.h5') model=keras.models.load_model(file)
试试打印信息:
成功了
总结
以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。