如何从Google Colab保存完整的Tensorflow模型

Question

我在google colab中训练了计算机视觉模型。

现在，我想将其保存到计算机上以进行操作。
如果我执行model.save("name.h5")然后下载.h5文件，是否保存了经过训练的模型？
还是仅仅是模型的未经训练的结构？

Answer 1

是的，mode.save("name.h5")保存已训练的模型。当然，您应该在训练/安装model之后执行此行。

但是，model.save()仅保存模型结构和更新的权重。并且它不存储任何损失函数权重和损失函数的信息。因此，从保存的文件加载模型后，应避免对模型进行重新训练。

如果您想进一步重新训练模型，则应使用tf.keras.models.save_model()保存模型，并使用tf.keras.models.load_model()加载模型。可以在Keras documentation中找到更多信息。谢谢。