Celery任务中导入keras模型

时间:2018-07-31 06:25:59

标签: tensorflow keras celery

我正在尝试使用芹菜中的tensorflow后端导入keras模型,但是该模型无法加载,并且什么也没有发生。我继续阅读其他文章,并尝试将import放入task函数,但这仍然无济于事。

代码看起来像这样:

@celery.task
def predict():
    from keras.models import load_model
    model = load_model(filepath)
    result = model.predict(data)
    return result

0 个答案:

没有答案