保存受Colab训练的模型以在AWS Sagemaker中进行部署

Question

我有一个Colab训练的模型。我想在AWS Sagemaker中部署它。为此，我尝试将经过训练的模型以protobuf格式保存在Colab中，但似乎仅以HDF5格式保存。我在Colab中尝试了TF 1.15和TF 2.2。从文档中，我了解到这是TF 2.2中的默认行为。当我遵循示例“ save_and_load.ipynb”时，它似乎起作用。

我的模型代码（TF 2.2）：

!mkdir -p saved_model
model.save('saved_model/my_model') 
!ls saved_model

输出：

/gdrive/My Drive/Machine Learning/out
my_model

示例代码（TF2.2）：

model.save('saved_model/my_model') 

my_model目录

！ls saved_model

包含资产文件夹，saved_model.pb和变量文件夹。

!ls saved_model/my_model

输出： my_model 资产save_model.pb变量