美文网首页
SKIL/工作流程/TensorFlow模型

SKIL/工作流程/TensorFlow模型

作者: hello风一样的男子 | 来源:发表于2019-04-19 17:04 被阅读0次

    TensorFlow模型

    SKIL1.0.3使用TensorFlow C++ API (版本1.4)的自定义JVM绑定,不施加限制。未来版本的SKIL将使用nd4j中的samediff功能来执行导入的模型。

    从TensorFlow导入
    参见导入模型页面。

    TensorFlow Word2Vec 支持

    如果使用了lookup_embedding函数,那么skil应该能够将模型导入到模型服务器中。

    部署的先决条件
    为了部署TensorFlow模型,图及其相关权重必须存储在单个pb文件中。使用tensorflow.python.tools下提供的freeze_graph.pyscript可以“冻结”TensorFlow模型。
    freeze_graph可以在命令行上运行,也可以在python脚本中运行。它需要一个图形定义和一组检查点,并将它们冻结在一个文件中,然后在skil中单击几下就可以部署这些文件。

    ~/tensorflow/bazel-bin/tensorflow/python/tools/freeze_graph \
      --input_graph=graph_definition.pbtxt \
      --output_graph=file_for_frozen_model.pb \
      --input_checkpoint=saved_ckpt \
      --output_nodes_names=output_node_name
    
    
    image.gif

    相关文章

      网友评论

          本文标题:SKIL/工作流程/TensorFlow模型

          本文链接:https://www.haomeiwen.com/subject/ygdcgqtx.html