【问题标题】:Will tf.loadFrozenModel and tf.loadModel have different prediction time?tf.loadFrozenModel 和 tf.loadModel 会有不同的预测时间吗?
【发布时间】:2019-02-17 08:46:59
【问题描述】:

假设在 tensorflow 中训练的模型使用两种方法将可用的模型转换为 tensorflowjs: 1) 使用 tf.saved_model.simple_save 方法将模型保存在 tensorflow 中,然后使用 tf.loadFrozenModel 将模型加载到 tensorflowjs 中并使用 model.predict 预测结果 2)使用keras(sequence)将模型保存到tensorflow中,然后使用tf.loadModel将模型加载到tensorflowjs中,使用model.predict预测结果

如果你在 tensorflow 中训练相同的模型,但保存方法不同。在tensorflowjs中使用上面2种加载模型的方法来预测结果,会不会有时间差?

【问题讨论】:

    标签: tensorflow tensorflow.js


    【解决方案1】:

    如果您在 tensorflowJs 和 keras 中具有相同的架构,则使用 tensorflowJs 的推理时间将是相似的。 tensorflowJs 转换器只会构建拓扑图和权重图。所以在这两种情况下处理时间大致相同

    【讨论】:

      猜你喜欢
      • 2022-01-14
      • 1970-01-01
      • 1970-01-01
      • 2020-11-05
      • 2021-09-19
      • 2021-10-06
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      相关资源
      最近更新 更多