Tensorflow2系類知識-7:部署以及並行計算

部署

模型導出

  • SaveModel:與前面介紹的 Checkpoint 不同,SavedModel 包含了一個 TensorFlow 程序的完整信息: 不僅包含參數的權值,還包含計算的流程(即計算圖) 。當模型導出爲 SavedModel 文件時,無需建立模型的源代碼即可再次運行模型,這使得 SavedModel 尤其適用於模型的分享和部署。後文的 TensorFlow Serving(服務器端部署模型)、TensorFlow Lite(移動端部署模型)以及 TensorFlow.js 都會用到這一格式。
tf.saved_model.save(model, "保存的目標文件夾名稱")
model = tf.saved_model.load("保存的目標文件夾名稱")

分佈式計算

單機 MirroredStrategy

  • tf.distribu
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章