『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。

 

注意,保存格式和保存方法並非一一對應,各接口支持多種保存格式。

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM