如何在Google Cloud Storage上保存经过训练的Tensorflow 2.0模型

Question

我正在按照本教程在Google Cloud中训练和部署Tensorflow 2.0模型：

https://github.com/GoogleCloudPlatform/data-science-on-gcp/blob/master/updates/cloudml/flights_model_tf2.ipynb

但是，当我尝试在Google Cloud Storage中保存经过训练的模型时，我总是会卡住。

我跑步时：

tf.keras.experimental.export_saved_model(model, export_dir)

我得到了错误：

TypeError： init （）缺少1个必需的位置参数：“ feature_columns”

该如何解决？我是tensorflow的新手。任何帮助将不胜感激。