keras_estimator.export_saved_model error: Expected keys: {'dense_3_input'}\n\tfeatures keys: {'dense_input'}

[wattli]

I followed this colab (https://colab.research.google.com/github/GoogleCloudPlatform/ml-on-gcp/blob/master/tutorials/explanations/ai-explanations-tabular.ipynb) but failed at the export_saved_model part.

Error msg: https://i.stack.imgur.com/gjTCd.png

Even I changed dense_input to dense_3_input, it still failed: https://i.stack.imgur.com/I2Aza.png

Details: https://stackoverflow.com/questions/62055398/keras-estimator-export-saved-model-error-expected-keys-dense-3-input-n-tfe/62056170#62056170