saved_model模塊主要用於TensorFlow Serving。TF Serving是一個將訓練好的模型部署至生產環境的系統,主要的優點在於可以保持Server端與API不變的情況下,部署新的算法或進行試驗,同時還有很高的性能。
在模型的訓練階段,同時要保存tfs需要的saved_model該如何去做呢,就是在你模型訓練已經完成,要保存成模型文件.pb或者.meta的形式,此時你就可以保存成tfs需要的文件格式.pbtxt,對應目錄下生成saved_model.pbtxt和variables,存儲網絡結構和對應的變量值。
如下圖: