模型节点操作学习笔记(1)--SavedModel详解

参考:使用 SavedModel 格式  |  TensorFlow Core (google.cn) (持续更新)  

  •  SavedModel是一个包含序列化签名运行这些签名所需的状态的目录,其中包含变量值和词汇表。
$ ls {mobilenet_save_path}
assets   fingerprint.pb   saved_model.pb   variables
  •  assets目录包含Tensorflow计算图使用的文件,例如,用于初始化词汇表的文本文件。
  •     saved_model.pb文件用于存储实际TensorFlow程序或者模型,以及一组已经命名的签名 ---- 每个签名标识一个接受张量输入产生张量输出的函数。

    variables 目录包含一个标准训练检查点

ls {mobilenet_save_path}/variables
variables.data-00000-of-00001  variables.index    
  • 加载和使用自定义模型;
    imported = tf.saved_model.load(module_no_signatures_path)
    assert imported(tf.constant(3.)).numpy() == 3
    imported.mutate(tf.constant(2.))
    assert imported(tf.constant(3.)).numpy() == 6
    

      

posted @ 2024-05-30 19:33  张幼安  阅读(10)  评论(0编辑  收藏  举报