遇到ValueError: Tensor Tensor(“dense_1/Sigmoid:0”, shape=(?, ?, 1), dtype=float32) is not an element …
ValueError: Tensor Tensor(“dense_1/Sigmoid:0”, shape=(?, ?, 1), dtype=float32) is not an element …
跑keras 和 tensorflow1.15版本的时候遇到的问题,记录一下
情景:使用keras跑h5模型进行predict时,出现的问题,可能是没有初始化
先关机重启
可能缓存出现混乱,我先重启了。
在代码一开始添加初始化
from keras import backend
import tensorflow as tf
global graph,model,sess
graph = tf.get_default_graph()
sess=backend.get_session()
加载模型
from keras.models import load_model
from att import Attention #自己的attention
model = load_model(“my_model.h5”, custom_objects={“Attention”: Attention})
重点!进行模型预测
with sess.as_default():
with graph.as_default():
# !重点就是添加上面两行!
predicted_ = model.predict(train_x)[0] #就是使用自己的模型进行预测