在TensorFlow中,可以使用tf.keras.models.save_model
和tf.keras.models.load_model
函数来保存和加载模型。
保存模型:
model.save("my_model.h5")
加载模型:
model = tf.keras.models.load_model("my_model.h5")
这样就可以将模型保存为一个.h5文件,并且在需要的时候加载模型。
辰迅云「云服务器」,即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘,价格低至29元/月。点击查看>>
推荐阅读: tensorflow中conv2d的作用是什么