python - 如何从 google colab 保存完整的 tensorflow 模型

Question

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

现在我想把它放到我的电脑上使用。
如果我执行 model.save("name.h5") 然后下载 .h5 文件，这会保存经过训练的模型吗？
或者它只是模型的未经训练的结构？

Answer 1

是的，model.save("name.h5") 保存训练好的模型。当然，您应该在训练/拟合模型后执行此行。

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

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