如何从 Colaboratory 中保存的检查点加载 TensorFlow Keras model？

Question

我在 Colaboratory 中构建并训练了一个 RNN model，并使用检查点回调将其保存：

tf.keras.callbacks.ModelCheckpoint( filepath=chkpt_p, save_weights_only=False)

保存的数据如下所示：

如果我没理解错的话，这个目录包括 model 和权重。 我应该使用哪个 function 来恢复它（在另一个运行时）？

Answer 1

这对我有用：

model = tf.keras.models.load_model(save_dir,compile=False)

我在使用tf.saved_model.load时遇到了一个问题，它不会产生 keras object，并且一个 TF 找不到损失 function 与compile=True 。