欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

Keras 多线程下运行的问题

程序员文章站 2022-05-02 18:07:36
...

用Keras模型进行预测,在多线程的情况下,特别是对于采用GPU时,会出现诸如类似以下的错误。

ValueError: Tensor Tensor("Sigmoid_2:0", shape=(?, 17), dtype=float32) is not an element of this graph.

解决方法如下:

        self.model.load_weights(weight_file_path)
        self.model._make_predict_function()
        self.graph = tf.get_default_graph()

在使用预测时 :

    with self.graph.as_default():
            predicted_class = np.argmax(self.model.predict(np.array([feature]))[0])