Python 如何使用Tensorflow估计器将带有权重的模型保存到单个protobuf文件中？

我正在尝试以

tf.saved\u model.save

生成的格式导出一个模型，它是一个大的.pb，包含整个模型和权重

我有

estimator.export_saved_model(
    export_dir_base=os.environ.get('SM_MODEL_DIR'),
    serving_input_receiver_fn=serving_input_fn
)

对于Sagemaker脚本模式培训脚本，但它似乎输出protobuf格式，带有一个大变量文件夹和一个小的.pb文件。如何让它以.pb格式导出完整模型

---

这是在Sagemaker培训实例上以脚本形式运行的，因此我希望尽可能避免在本地执行手动冻结_graph.py命令。

Hi@King Dedede，请提供有关文件的更多详细信息或屏幕截图，如果您还可以提供最低可复制代码。

Tensorflow

，无法使用

tf.saved_Model.save

或

tf.keras.models.save_Model

将模型保存在单个protobuf文件中。您可以检查此链接。谢谢学习愉快！您好@King Dedede，请提供有关文件的更多详细信息或屏幕截图，如果您还可以提供最小可复制代码。在

Tensorflow

中，使用

tf.saved_Model.save

或使用

tf.keras.models.save_Model

无法将模型保存在单个protobuf文件中。您可以检查此链接。谢谢学习愉快！