tensorflow怎么保存和加载模型

lewis 6年前 (2020-01-21) 阅读数 10 #大数据

文章标签 tensorflow

在TensorFlow中，可以使用tf.keras.models.save_model和tf.keras.models.load_model函数来保存和加载模型。

保存模型：

model.save("my_model.h5")

加载模型：

model = tf.keras.models.load_model("my_model.h5")

这样就可以将模型保存为一个.h5文件，并且在需要的时候加载模型。

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

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