【发布时间】：2020-07-10 10:17:20
【问题描述】：
我在 google colab 中训练了一个计算机视觉模型。
现在我想把它放到我的电脑上使用它。
如果我做model.save("name.h5") 然后下载.h5 文件，这会保存训练好的模型吗？
还是只是模型未经训练的结构？
【问题讨论】：
标签： python tensorflow keras deep-learning google-colaboratory
【发布时间】：2020-07-10 10:17:20
【问题描述】：
是的，model.save("name.h5") 保存了训练好的模型。当然，您应该在训练/拟合模型后执行此行。
但是，model.save() 只保存模型结构和更新的权重。并且它不存储任何损失函数的权重和损失函数的信息。因此，您应避免在从保存的文件加载模型后重新训练模型。
如果您愿意进一步重新训练您的模型，您应该使用tf.keras.models.save_model() 保存模型，并使用tf.keras.models.load_model() 加载模型。更多信息可以在 Keras documentation 中找到。
【讨论】：
-
“而且它不存储任何权重”这看起来不对，它与您之前所说的相矛盾。 model.save 确实保存了模型的权重。
-
对不起，我想我写的方式让你误解了。因此，我编辑了答案。我想进一步补充一些信息。有时 Keras 损失函数包含用于训练模型的重要权重。使用
model.save()不会保存损失函数权重。但是，如果您不愿意对保存的模型进行预训练，这不会造成任何问题。谢谢。