我正在使用带有Tensorflow的Keras作为后端.
我试图在我的主进程中保存模型,然后model.predict在另一个进程中加载/运行(即调用).
我目前只是尝试从文档中保存/加载模型的天真方法:https://keras.io/getting-started/faq/#how-can-i-save-a-keras-model.
所以基本上:
model.save() 在主要过程中model = load_model() 在儿童过程中model.predict() 在儿童过程中但是,它只是挂在load_model电话上.
搜索周围我发现了这个可能相关的答案,表明Keras只能在一个过程中使用:与theano一起使用多处理但不确定这是否属实(似乎在这方面似乎找不到太多).
有没有办法实现我的目标?非常感谢高级别描述或简短示例.
注意:我尝试了将图形传递给流程的方法但是失败了,因为似乎张流图不可选(这里有相关的SO帖子:Tensorflow:将会话传递给python多进程).如果确实有一种方法可以将张量流图/模型传递给子进程,那么我也对此持开放态度.
谢谢!
python neural-network python-multiprocessing keras tensorflow