.pb SavedModel和.tf SavedModel有什么区别？

Question

For.pb SavedModel： model.save("my_model") 默认保存为.pb

对于.tf SavedModel： model.save("my_model",save_format='.tf')

我想知道这两种格式之间的区别。他们都是SavedModel吗？他们俩都一样吗？哪个更好？两者都是TensorFlow扩展吗？

Answer 1

请参阅tf.keras.Model.save的文档。 save_format可以具有两个值之一：

• tf（在TensorFlow 2.x中为默认值）表示TensorFlow格式，一种SavedModel协议缓冲文件。
• h5（在TensorFlow 1.x中为默认值）表示HDF5 Keras格式，在Keras完全独立于TensorFlow并定义为支持多个后端而不特别绑定任何对象时定义。

在TensorFlow 2.x中，您永远不需要h5，除非您要生成与旧版本或类似版本兼容的文件。还可以将SavedModel集成到TensorFlow生态系统中，例如，如果您想将其与TensorFlow Serving一起使用。