tensorflow2.x学习笔记九：保存和加载模型

一、模型的保存

我们在保存模型的时候既可以保存权重+图结构，也可以只保存权重

1. 保存 权重+图结构：

有四种方式，一种是直接调用saved_model下的save API；另一种是添加callbacks参数，后者要注意save_weights_only 参数要设为False；还有一种就是直接使用model.save API；最后一种就是使用tf.keras.models下的save_model API。

tf.saved_model.save，保存的文件格式是.pb，传入的参数是文件所在文件夹的路径

tf.saved_model.save(model, 
                    os.path.join("graph_and_weights"))

设置callbacks参数，保存的是.h5格式的文件