我想在我的GPU上运行这个Keras模型,但是它运行在我的cpu上,我用Plaidml来使用我的GPU,plaidml是正确设置的,它运行在其他的模型上--我想也许我是在导入tensoflow,但我不确定,我需要在GPU上运行这个模型,我有其他不导入tensorflow的模型,它们与Plaidml很好地工作。源代码
from os impo
我有一些预先训练好的二进制Keras模型。在此示例中为3个模型。我试图通过平均这些模型的预测来建立一个集成模型。为了做到这一点,我引用了集成模型中的示例。我的代码如下for i in os.listdir(model_root): # to get all the models in one list models.append(model)
# e