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

Posted 2020-12-14 farwish

 

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