【问题标题】:How to load TensorFlow Keras model from a saved checkpoint in Colaboratory?如何从 Colaboratory 中保存的检查点加载 TensorFlow Keras 模型?
【发布时间】:2020-11-15 13:15:09
【问题描述】:
我在 Colaboratory 中构建并训练了一个 RNN 模型,并使用检查点回调将其保存:
tf.keras.callbacks.ModelCheckpoint( filepath=chkpt_p, save_weights_only=False)
保存的数据如下所示:
如果我理解正确的话,这个目录包括模型和权重。我应该使用哪个函数来恢复它(在另一个运行时)?
【问题讨论】:
标签:
python
tensorflow
keras
google-colaboratory
【解决方案1】:
这对我有用:
model = tf.keras.models.load_model(save_dir,compile=False)
我在使用 tf.saved_model.load 时遇到了一个问题,它不会生成 keras 对象,并且一个 TF 没有使用 compile=True 找到损失函数。