tensorflow – Keras花了很长时间在model.load()之后进行第一次预测

我正在使用model.load()在keras中加载模型,并且我发现第一个预测的计算时间比预测的时间长10倍以上,任何想法可能会发生这种情况或建议使负载初始化 – 首先预测周期加快将非常感激.

我正在使用Tensorflow后端进行CPU处理.

谢谢您的帮助,
Denym

最佳答案 好的,我找到了适合我的答案:

如果您同时加载许多模型,请不要使用keras model.load函数,将结构保存为json / yaml,将权重保存为.h5并按照keras示例加载.

处理少于5个模型时,model.load函数要快得多,但加载时间会以指数方式增加同时加载的模型.

当从加载结构和权重方法加载结构和权重方法时,从json加载和从.h5加权的速度快10倍,虽然每个模型有一些减速,但它是线性的而不是指数的,这使得它在一次加载多个模型时显着更快.

点赞