TensorFlow中怎么保存和加载模型

lewis 1年前 (2024-04-13) 阅读数 11 #大数据

文章标签 tensorflow

在 TensorFlow 中，可以使用 tf.keras.models.save_model() 方法保存模型，使用 tf.keras.models.load_model() 方法加载模型。

保存模型示例代码：

model.save('path/to/save/model.h5')

加载模型示例代码：

model = tf.keras.models.load_model('path/to/save/model.h5')

本文仅代表作者观点，不代表米安网络立场。

◎欢迎参与讨论，请在这里发表您的看法、交流您的观点。