部署
模型導出
- 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