在TensorFlow中,可以使用tf.saved_model.save()方法来导出模型。具体步骤如下:
创建并训练模型。
使用tf.saved_model.save()方法保存模型。例如:
model.save('path_to_saved_model')
tf.saved_model.save(model, 'path_to_saved_model', signatures={'serving_default': predict})
辰迅云「云服务器」,即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘,价格低至29元/月。点击查看>>
推荐阅读: TensorFlow中的模型融合方法有哪些