一、模型的保存
我们在保存模型的时候既可以保存权重+图结构,也可以只保存权重
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格式的文件</