如何从谷歌colab保存完整的张量流模型

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文档中找到。