参考:使用 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