tensorflow怎么保存和加载模型
lewis
2020-01-21
19次阅读
在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文件,并且在需要的时候加载模型。

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