TensorFlow模型保存

需求

辛苦训练了一晚的模型，想保存下来分享给小伙伴使用。

方法

保存权重

可以选择保存模型的权重文件为checkpoint，但下次复现时需要搭建相同的主干网络，不推荐。

model.save_weights('./checkpoints/epoch_100')

保存模型

可以脱离原来的代码，适合分享模型无需再次搭建网络。

# save
model.save('./saved_model/epoch_100')
# load
tf.keras.models.load_model('./saved_model/epoch_100')

参考

保存和恢复模型  |  TensorFlow Core