【问题标题】: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 找到损失函数。

    【讨论】:

      猜你喜欢
      • 1970-01-01
      • 2019-04-12
      • 2019-03-30
      • 1970-01-01
      • 2021-03-08
      • 2021-04-21
      • 2019-08-28
      • 1970-01-01
      • 2020-02-29
      相关资源
      最近更新 更多