2.5 模型保存与加载

主要有三种方法：

• model.save_weights() : 该方法可保存模型权重；

• model.save() :该方法可保存HDF5模型文件，或者pb格式文件；

• tf.keras.models.save() : 该方法可保存HDF5模型文件，或者pb格式文件；

• tf.saved_model.save() : 该方法可保存pb格式文件，主要用于模型服务。

• model.load_weights() : 加载模型权重，和model.save_weights配合使用。

• tf.keras.models.load_model() : 加载HDF5或pb模型文件，和model.save()、tf.keras.models.save_models()配合使用

• tf.saved_model.load() :加载模型文件，和tf.saved_model.save()配合使用

小知识

tf.keras.models.save、model.save 和 tf.saved_model.save()保存的pb文件有何区别？

tf.saved_model格式的模型可以直接用来预测(predict)，但是 tf.saved_model没有保存优化器配置。 反之，tf.keras.models.save()保存模型的所有参数，包含优化器等参数，可直接进行模型预测(model.predict())。