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

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

Refer:TensorFlow流行度

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

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