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])
上一篇: 定时任务开多线程
下一篇: pyquery学习笔记