标签: python tensorflow keras multiprocessing
我正在尝试在Keras Lstm模型中实现多处理。
我尝试过model.predict(data,use_multiprocessing = True,worker = 4)。 尽管我的机器有足够的资源,但这并不能提高速度。
我也尝试过multiprocessing.pool。 我收到一个与数据形状有关的错误,它是二维张量。由于某种原因,pool.map()无法识别输入形状,尽管它在model.predict中可以正常工作。
任何建议都值得赞赏!