标签: tensorflow keras gpu tensorflow2.0
在多标签分类的情况下,我用 Keras 训练了 5 个分类模型。每一个都致力于识别所有其他类别中的一个类别(二元分类),然后我将每个模型的所有预测结果连接起来,以获得5个类别的最终分类向量。
对于推理,有没有办法在同一图像上并行运行这 5 个模型,以加快过程? (我可以使用多个 GPU)。然后我会得到所有的预测结果并将它们连接起来。
非常感谢,