ValueError: Tensor Tensor(“Placeholder:0“, shape=(,), dtype=float32) is not an element of this graph

fastapi + tf

再用docker打包项目时，镜像运行出以上错误。百度了下发现是web应用的模型初始化和模型调用初始化不在一个线程。不同graph。

反正云里雾里，不太了解这个，就照着改了，改了半下午，在此记录下，给自己长记性

from tensorflow.python.keras.backend import set_session 

在load model前声明session，graph

set_session(sess)

每次在使用tf时，

global sess

global graph

with graph.as_default():

        set_session(sess)

        model.predict(...)