时间戳

首页 > 解决方案 > 将模型保存为 cnn.model 或 cnn.h5 有什么区别？这些扩展有什么不同？

keras - 将模型保存为 cnn.model 或 cnn.h5 有什么区别？这些扩展有什么不同？

问题描述

我正在使用model.save("cnn.model")并model.save("cnn.h5")在训练后保存模型。

在 2 个不同的扩展中保存模型有什么区别？

标签： kerasdeep-learning

解决方案

包含扩展名的文件名无关紧要。不管是什么，Keras 都会将 HDF5 格式的模型保存到该文件中。

Doc：如何保存 Keras 模型？

您可以使用model.save(filepath)将 Keras 模型保存到单个 HDF5 文件中，该文件将包含：

  • 模型的架构，允许重新创建模型
  • 模型的权重
  • 训练配置（损失、优化器）
  • 优化器的状态，允许在您停止的地方恢复训练。

推荐阅读