1. 程式人生 > 實用技巧 >『TensorFlow』TF2的模型儲存

『TensorFlow』TF2的模型儲存

官網指南:https://www.tensorflow.org/tutorials/keras/save_and_load

簡單總結:

僅儲存權重有兩種方法:keras的fit介面可以用回撥函式實現,也可以用model.save_weight實現;同時支援兩種格式:ckeckpoint即tf1中的經典格式,還支援HDF5格式。

儲存整個模型有兩種方法:tf.saved_model.save是正統的儲存模型介面;model.save也可以儲存整個模型;同樣支援兩種格式:SavedModel和HDF5,SavedModel是tf server支援的格式,也就是說部署是偏工業的格式。指南最後提到了兩種格式的差異:HDF5使用物件配置儲存模型結構,而 SavedModel 儲存執行圖。因此,SavedModel 能夠儲存自定義物件,例如子類化模型和自定義層,而無需原始程式碼,至於HDF5怎麼儲存自定義結構自行參閱指南。更多SavedModel格式介紹:

https://tensorflow.google.cn/guide/saved_model

載入整個模型的介面為:tf.keras.models.load_model。

注意,儲存格式和儲存方法並非一一對應,各介面支援多種儲存格式。