[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