Keras模型中的多处理预测

时间:2019-11-19 21:06:00

标签: python tensorflow keras multiprocessing

我正在尝试在Keras Lstm模型中实现多处理。

我尝试过model.predict(data,use_multiprocessing = True,worker = 4)。 尽管我的机器有足够的资源,但这并不能提高速度。

我也尝试过multiprocessing.pool。 我收到一个与数据形状有关的错误,它是二维张量。由于某种原因,pool.map()无法识别输入形状,尽管它在model.predict中可以正常工作。

任何建议都值得赞赏!

0 个答案:

没有答案