遇到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 …

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] #就是使用自己的模型进行预测