【发布时间】：2019-06-14 12:58:42
【问题描述】：
我希望为 Tensorflow 模型提供服务，以制作 Docker 映像并使用 AWS 进行部署。为此，我需要在为任何 Tensorflow 模型提供服务时必须拥有 .pb 和变量文件。但是，我只有模型的检查点文件。有没有办法从检查点文件中恢复变量文件夹？
我可以创建 .pb 文件，但不确定如何获取变量文件夹。
【问题讨论】：
标签： python tensorflow deep-learning
【发布时间】：2019-06-14 12:58:42
【问题描述】：
ckpt = tf.train.latest_checkpoint(args.model_path)
model.load_weights(ckpt)
ckpt_filename = os.path.basename(ckpt)
saved_model_path = os.path.join('pb_files', ckpt_filename)
model.save(saved_model_path)
https://www.tensorflow.org/guide/saved_model
你好，我从上面的文档中创建了这个 sn-p。此代码将创建 pb 文件、变量文件夹和资产文件夹。
【讨论】：