码迷,mamicode.com
首页 > 其他好文 > 详细

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

时间:2020-07-03 12:28:49      阅读:189      评论:0      收藏:0      [点我收藏+]

标签:tutorial   copy   distrib   创建   相同   html   xxxxx   arw   imp   

 

将模型保存为完整的 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

标签:tutorial   copy   distrib   创建   相同   html   xxxxx   arw   imp   

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

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!