如何从TensorFlow中的检查点获取变量和.pb文件?

时间:2019-06-14 12:58:42

标签: python tensorflow deep-learning

我正在寻找服务Tensorflow模型来制作Docker映像并使用AWS进行部署。为此,我需要拥有在服务任何Tensorflow模型时必须的.pb和变量文件。但是,我只有模型的检查点文件。有什么办法可以从检查点文件中还原变量文件夹?

我能够创建.pb文件,但不确定如何获取变量文件夹。

1 个答案:

答案 0 :(得分:0)

RewriteEngine On
ReWriteRule ^detail/([a-zA-Z]+)$ detail.php?v=$1

https://www.tensorflow.org/guide/saved_model

您好,我是根据上述文档创建的代码段。该代码将创建pb文件,变量文件夹和资产文件夹。