使用 keras/Tensorflow 2.3 并行运行多个模型

时间:2021-02-10 11:43:07

标签: tensorflow keras gpu tensorflow2.0

在多标签分类的情况下,我用 Keras 训练了 5 个分类模型。每一个都致力于识别所有其他类别中的一个类别(二元分类),然后我将每个模型的所有预测结果连接起来,以获得5个类别的最终分类向量。

对于推理,有没有办法在同一图像上并行运行这 5 个模型,以加快过程? (我可以使用多个 GPU)。然后我会得到所有的预测结果并将它们连接起来。

非常感谢,

0 个答案:

没有答案