当前位置: 代码迷 >> 综合 >> [Keras] ValueError: Tensor Tensor is not an element of this graph.[Solved]
  详细解决方案

[Keras] ValueError: Tensor Tensor is not an element of this graph.[Solved]

热度:51   发布时间:2023-12-12 01:02:28.0

自己使用load_weights函数加载预测时正常,但利用用flask使用model.load_weights函数加载keras模型进行预测时,出现的ValueError: Tensor Tensor(“loss/mul:0”, shape=(), dtype=float32) is not an element of this graph.的错误。

采用gevent启动服务:

from gevent.pywsgi import WSGIServer## flask server# app.debug=True# app.run(host='0.0.0.0')## Serve the app with geventhttp_server = WSGIServer(('',5000),app)http_server.serve_forever()

即可解决!

参考
  • https://github.com/keras-team/keras/issues/2397
  相关解决方案