文章/答案/技术大牛

发布

社区首页 >问答首页 >为什么导入tensorflow.keras库的推理时间比导入keras库的推理时间长？

问为什么导入tensorflow.keras库的推理时间比导入keras库的推理时间长？
EN

Stack Overflow用户

提问于 2020-06-25 03:17:35

回答 1查看 52关注 0票数 0

问题是在推理时使用tensorflow.keras.models.load_model加载网络模型时会有很大的差异。如果我使用keras.models.load_model加载模型，推理速度比tensorflow.keras库快3倍。

模型存储在.h5文件中。可能只有一件事是“错误的”：首先我使用keras库训练并保存了模型，但(由于代码演变后的其他导入冲突)现在我使用tensorflow.keras加载它。

库版本包括：

tensorflow==2.2.0
tensorflow-estimator==2.2.0
tensorflow-metadata==0.22.0
Keras==2.3.1
Keras-Applications==1.0.8
Keras-Preprocessing==1.1.1

发布于 2020-06-25 03:20:47

是的，我也遇到过这个问题。

当您使用keras进行训练并使用tensorflow.keras进行加载时，就会发生这种情况。

为了解决您的问题，请确保您正在使用tensorflow.keras进行训练和推理

票数 0

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/62562535

复制

相似问题

问为什么导入tensorflow.keras库的推理时间比导入keras库的推理时间长？EN