tensorflow serving多模型部署

本文档详细介绍了如何使用TensorFlow Serving进行多模型部署,包括从Docker镜像下载、模型格式转换、配置文件编写到服务启动的全过程。重点讲解了kashgari框架如何生成可供TensorFlow Serving读取的pb模型,并提供了CPU和GPU版本的启动命令。

基本流程

  • docker pull tensorflow serving官方镜像
  • 生成tensorflow serving可直接读取的模型格式
  • 编写配置文件
  • 启动tensorflow serving镜像

1、docker下载

  • CPU版本
docker pull tensorflow/serving:latest
  • GPU版本
docker pull tensorflow/serving:latest-gpu

2、模型格式转换

tensorflow serving支持pb模型,部署多模型所需的相关模型结构目录如下:

在这里插入图片描述

我用的是kashgari框架,其提供直接保存pb模型的方法kashgari.utils.convert_to_saved_model


 -----model.save:保存在reberate_classification_13文件夹下的h5模型

 -----convert_to_saved

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值