首页 > 其他 > 详细

[ML] Tensorflow2 保存完整模型以及使用 HDF5

时间:2020-07-03 14:03:29      阅读:150      评论:0      收藏:0      [点我收藏+]

 

将模型保存为完整的 HDF5 文件,后面可以直接加载使用:

# © cnblogs.com/farwish
import tenforflow as tf

model = tf.keras.models.xxxxx
model.compile(xxx)
model.fit(xxx)

import time
saved_model_path = "./saved_models/{}.h5".format(int(time.time()))

model.save(saved_model_path)

 

重新创建完全相同的模型,包括其权重和优化程序:

# © cnblogs.com/farwish
import tensorflow as tf

new_model = tf.keras.models.load_model(saved_model_path)

 

Ref:https://tensorflow.google.cn/tutorials/distribute/save_and_load

Link:https://www.cnblogs.com/farwish/p/13229491.html

[ML] Tensorflow2 保存完整模型以及使用 HDF5

原文:https://www.cnblogs.com/farwish/p/13229491.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!