DEPLOY.md 32.1 KB
Newer Older
X
xulongteng 已提交
1
# PaddlePaddle分布式训练和Serving流程化部署
X
xulongteng 已提交
2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17

* [ 环境配置](#head0)
* [ 分布式训练](#head1)
	* [ 1、集群配置](#head2)
		* [1.1 创建集群](#head3)
		* [1.2 配置集群环境](#head4)
	* [2、 配置开发机环境](#head5)
		* [2.1 安装KubeCtl](#head6)
		* [2.2 安装Helm](#head7)
		* [2.3 配置文件](#head8)
		* [2.4 安装Go](#head9)
	* [ 3、安装volcano](#head10)
	* [ 4、执行训练](#head11)
	* [ 5、模型产出](#head12)
		* [5.1 模型裁剪](#head13)
		* [5.2 稀疏参数产出](#head14)
X
xulongteng 已提交
18 19 20 21 22 23 24 25 26 27
* [ 大规模稀疏参数服务Cube的部署和使用](#head15)
	* [1. 编译](#head16)
	* [2. 分片cube server部署](#head17)
		* [2.1 配置文件修改](#head18)
		* [2.2 拷贝可执行文件和配置文件到物理机](#head19)
		* [2.3 启动 cube server](#head20)
	* [3. cube-builder部署](#head21)
		* [3.1 配置文件修改](#head22)
		* [3.2 拷贝可执行文件到物理机](#head23)
		* [3.3 启动cube-builder](#head24)
D
Dang Yifei 已提交
28
		* [3.4 seqfile工具](#head241)
X
xulongteng 已提交
29 30 31 32
	* [4. cube-transfer部署](#head25)
		* [4.1 cube-transfer配置修改](#head26)
		* [4.2 拷贝cube-transfer到物理机](#head27)
		* [4.3 启动cube-transfer](#head28)
D
Dang Yifei 已提交
33
		* [4.4 cube-transfer支持查询接口](#head281)
X
fix toc  
xulongteng 已提交
34
        * [4.5 donefile格式协议](#head29)
X
xulongteng 已提交
35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59
* [ 预测服务部署](#head30)
	* [ 1、Server端](#head31)
		* [1.1 Cube服务](#head32)
		* [1.2 Serving编译](#head33)
		* [1.3 配置修改](#head34)
			* [1.3.1 conf/gflags.conf](#head35)
			* [1.3.2 conf/model_toolkit.prototxt](#head36)
			* [1.3.3 conf/cube.conf](#head37)
			* [1.3.4 模型文件](#head38)
		* [1.4 启动Serving](#head39)
	* [ 2、Client端](#head40)
		* [2.1 测试数据](#head41)
		* [2.2 Client编译与部署](#head42)
			* [2.2.1 配置修改](#head43)
			* [2.2.2 运行服务](#head44)

在搜索、推荐、在线广告等业务场景中,embedding参数的规模常常非常庞大,达到数百GB甚至T级别;训练如此规模的模型需要用到多机分布式训练能力,将参数分片更新和保存;另一方面,训练好的模型,要应用于在线业务,也难以单机加载。Paddle Serving提供大规模稀疏参数读写服务,用户可以方便地将超大规模的稀疏参数以kv形式托管到参数服务,在线预测只需将所需要的参数子集从参数服务读取回来,再执行后续的预测流程。

本文以CTR预估任务为例,提供一个完整的基于PaddlePaddle的分布式训练和Serving的流程化部署过程。基于此流程,用户可定制自己的端到端深度学习训练和应用解决方案。

本文演示的基于PaddlePaddle的分布式训练和Serving流程化部署,基于CTR预估任务,原始模型可参见[PaddlePaddle公开模型github repo](https://github.com/PaddlePaddle/models/tree/develop/PaddleRec/ctr)。 整体拓扑架构如下图所示:

![PaddlePaddle分布式训练和Serving流程化部署拓扑](./deploy/ctr-prediction-end-to-end-deployment.png)

其中:
W
Wang Guibao 已提交
60 61 62 63
1. 分布式训练集群在百度云k8s集群上搭建,并通过[volcano](https://volcano.sh/)提交分布式训练任务和资源管理
2. 分布式训练产出dense参数和ProgramDesc,通过http服务直接下载到Serving端,给Serving加载
3. 分布式训练产出sparse embedding,由于体积太大,通过cube稀疏参数服务提供给serving访问
4. 在线预测时,Serving通过访问cube集群获取embedding数据,与dense参数配合完成预测计算过程
X
xulongteng 已提交
64

W
Wang Guibao 已提交
65
以下从4部分分别介绍上图中各个组件:
W
Wang Guibao 已提交
66 67 68 69
1. 分布式训练集群和训练任务提交
2. 稀疏参数服务部署与使用
3. Paddle Serving的部署
4. 客户端访问Paddle Serving完成CTR预估任务预测请求
X
xulongteng 已提交
70
## <span id="head0"> 环境配置</span>
X
add doc  
xulongteng 已提交
71 72 73

**环境要求** :helm、kubectl、go

X
xulongteng 已提交
74
## <span id="head1"> 分布式训练</span>
X
add doc  
xulongteng 已提交
75 76 77

分布式训练采用[volcano](https://github.com/volcano-sh/volcano)开源框架以及云平台实现,文档中以[百度智能云](https://cloud.baidu.com/?from=console)以及CTR预估模型为例,演示如何实现大规模稀疏参数模型的分布式训练。

X
xulongteng 已提交
78
### <span id="head2"> 1、集群配置</span>
X
add doc  
xulongteng 已提交
79

X
xulongteng 已提交
80
#### <span id="head3">1.1 创建集群</span>
X
add doc  
xulongteng 已提交
81 82 83

登录百度智能云官网,参考[帮助文档](https://cloud.baidu.com/doc/CCE/s/zjxpoqohb)创建容器引擎。

X
xulongteng 已提交
84
#### <span id="head4">1.2 配置集群环境</span>
X
add doc  
xulongteng 已提交
85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105

进入“产品服务>容器引擎CCE”,点击“集群管理>集群列表”,可看到用户已创建的集群列表。从集群列表中查看创建的集群信息。

![img](./deploy/cluster-info.png)

点击左侧的"Helm>Helm实例",点击安装链接为集群一键安装helm。百度智能云为集群安装的helm版本为2.12.3,kubectl版本为1.13.4

为了能够从外部登录集群节点,需要为集群中安装了tiller的节点申请弹性公网。点击"更多操作>控制台"。

![concole](./deploy/concole.png)

点击"命名空间"选择kube-system,点击"容器组",查看tiller开头的节点。

![tiller](./deploy/tiller.png)

点击"产品服务>网络>弹性公网"

![eip](./deploy/eip.png)

创建弹性公网实例,完成后选择创建的实例,点击"更多操作>绑定到BCC",填写tiller开头的节点信息进行绑定。

X
xulongteng 已提交
106
### <span id="head5">2、 配置开发机环境</span>
X
add doc  
xulongteng 已提交
107 108 109

配置过程需要开发机的root权限。

X
xulongteng 已提交
110
#### <span id="head6">2.1 安装KubeCtl</span>
X
add doc  
xulongteng 已提交
111 112 113

KubeCtl可以实现在本地开发机上连接百度智能云的Kubernets集群,建议参考百度云操作指南文档中[通过KubeCtl连接集群](https://cloud.baidu.com/doc/CCE/s/6jxpotcn5)部分进行安装。

X
xulongteng 已提交
114
#### <span id="head7">2.2 安装Helm</span>
X
add doc  
xulongteng 已提交
115 116 117 118 119 120 121

建议参考[Helm官方安装文档](https://helm.sh/docs/using_helm/#installing-helm)进行安装。

**注意事项:**

开发机上的kubectl与helm的版本需要与集群上的版本相一致,目前百度智能云为集群安装的helm版本为2.12.3,kubectl版本为1.13.4。

X
xulongteng 已提交
122
#### <span id="head8">2.3 配置文件</span>
X
add doc  
xulongteng 已提交
123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161

点击"集群列表"界面的"配置文件下载",下载配置文件。

![conf download](./deploy/conf-download.png)

将下载的配置文件移动到~/.kube文件夹下,文件名修改为config。

通过之前创建的弹性公网ip登录运行tiller的节点,账户密码为创建集群时设置的账户和密码,默认账户为root。

将节点上的以下三个文件

> /etc/kubernetes/pki/ca.pem
>
> /etc/kubernetes/pki/admin.pem
>
> /etc/kubernetes/pki/admin-key.pem

下载至开发机并放在相同的路径,添加四个环境变量

```bash
export HELM_TLS_ENABLE=true

export HELM_TLS_CA_CERT=/etc/kubernetes/pki/ca.pem

export HELM_TLS_CERT=/etc/kubernetes/pki/admin.pem

export HELM_TLS_KEY=/etc/kubernetes/pki/admin-key.pem
```

分别执行`kubectl version``helm version`,如果返回client端与server端信息,则证明配置成功。

示例:

![kubectl version](./deploy/kubectl-version.png)

![helm version](./deploy/helm-version.png)

如果只返回client端信息,server端信息显示"Forbidden",检查开发机是否使用了代理,若有可以尝试关闭代理再次执行命令检查。

X
xulongteng 已提交
162
#### <span id="head9">2.4 安装Go</span>
X
add doc  
xulongteng 已提交
163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183

推荐安装Go 1.12

下载安装包

```bash
wget https://studygolang.com/dl/golang/go1.12.7.linux-amd64.tar.gz --no-check-certificate
```

解压到 /usr/local/路径下

```bash
tar zxvf go1.12.7.linux-amd64.tar.gz -C /usr/local/
```

设置环境变量

```bash
export GOPATH=/usr/local/go
```

X
xulongteng 已提交
184
### <span id="head10"> 3、安装volcano</span>
X
add doc  
xulongteng 已提交
185 186 187 188 189 190 191 192 193 194 195 196 197 198 199

参考[volcano官方文档](https://github.com/volcano-sh/volcano#quick-start-guide)

通过yaml文件安装

```bash
kubectl apply -f https://raw.githubusercontent.com/volcano-sh/volcano/master/installer/volcano-development.yaml
```

安装完成后执行`kubectl get pods --namespace volcano-system`

若出现以下信息则证明安装成功:

![volcano](./deploy/volcano.png)

X
xulongteng 已提交
200
### <span id="head11"> 4、执行训练</span>
X
add doc  
xulongteng 已提交
201 202 203 204 205 206 207

创建cluster role和service account,[defaultserviceaccountclusterrole.yaml](./resource/defaultserviceaccountclusterrole.yaml) 文件示例如下:

```yaml
kind: ClusterRole
apiVersion: rbac.authorization.k8s.io/v1
metadata:
X
xulongteng 已提交
208 209
name: default
namespace: default
X
add doc  
xulongteng 已提交
210
rules:
X
xulongteng 已提交
211 212 213
- apiGroups: [""]
resources: ["pods"]
verbs: ["get", "list", "watch"]
X
add doc  
xulongteng 已提交
214 215 216 217 218

---
kind: ClusterRoleBinding
apiVersion: rbac.authorization.k8s.io/v1
metadata:
X
xulongteng 已提交
219 220
name: default
namespace: default
X
add doc  
xulongteng 已提交
221
subjects:
X
xulongteng 已提交
222 223 224
- kind: ServiceAccount
name: default
namespace: default
X
add doc  
xulongteng 已提交
225
roleRef:
X
xulongteng 已提交
226 227 228
kind: ClusterRole
name: default
apiGroup: rbac.authorization.k8s.io
X
add doc  
xulongteng 已提交
229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256
```

执行

```bash
kubectl create -f defaultserviceaccountclusterrole.yaml 
```

CTR模型的训练镜像存放在[dockerhub](https://hub.docker.com/)网站,通过kubectl加载yaml文件启动训练任务,CTR预估模型训练任务的yaml文件为[volcano-ctr-demo-baiduyun.yaml](./resource/volcano-ctr-demo-baiduyun.yaml)

执行

```bash
kubectl apply -f volcano-ctr-demo-baiduyun.yaml
```

通过`kubectl get pods`命令可以查看训练任务的运行情况

![ctr running](./deploy/ctr-running.png)

通过`kubectl logs $POD_NAME`可以查看对应的日志,例如`kubectl logs edl-demo-trainer-0`

![trainer log](./deploy/trainer-log.png)

也可以通过百度云平台提供的web页面观察集群的工作负载

![工作负载](./deploy/workload.png)

X
xulongteng 已提交
257
### <span id="head12"> 5、模型产出</span>
X
add doc  
xulongteng 已提交
258

W
Wang Guibao 已提交
259
CTR预估模型包含了embedding部分以及dense神经网络两部分,其中embedding部分包含的稀疏参数较多,在某些场景下单机的资源难以加载整个模型,因此需要将这两部分分割开来,稀疏参数部分放在分布式的稀疏参数服务,dense网络部分加载到serving服务中。在本文中使用的CTR模型训练镜像中已经包含了模型裁剪和稀疏参数产出的脚本,以下简述其原理和工作过程。
X
add doc  
xulongteng 已提交
260

W
Wang Guibao 已提交
261
#### <span id="head13">5.1 模型裁剪,产出预测ProgramDesc和dense参数</span>
X
add doc  
xulongteng 已提交
262

W
Wang Guibao 已提交
263
产出用于paddle serving预测服务的dense模型需要对保存的原始模型进行裁剪操作,修改模型的输入以及内部结构。具体原理和操作流程请参考文档[改造CTR预估模型用于大规模稀疏参数服务演示](https://github.com/PaddlePaddle/Serving/blob/develop/doc/CTR_PREDICTION.md)
X
add doc  
xulongteng 已提交
264

W
Wang Guibao 已提交
265
在trainer镜像中,模型裁剪的主要交互流程是:
X
add doc  
xulongteng 已提交
266

W
Wang Guibao 已提交
267 268 269 270
1. 监视训练脚本所在目录的models文件夹,当发现有子目录`pass-1000`时,表示训练任务完成 (默认训练轮次为1000)
2. 调用save_program.py,生成一个适用于预测的ProgramDesc保存到models/inference_only目录,并将所需参数一并保存到该子目录下
3. 调用replace_params.py,用models/pass-1000目录下参数文件替换models/inference_only目录下同名参数文件
4. 打包models/inference_only生成ctr_model.tar.gz,放到HTTP服务目录下,供外部用户手动下载,并替换到Serving的data/models/paddle/fluid/ctr_prediction目录中 (详见本文“预测服务部署”一节)
X
add doc  
xulongteng 已提交
271

W
Wang Guibao 已提交
272
#### <span id="head14">5.2 稀疏参数产出</span>
X
add doc  
xulongteng 已提交
273

W
Wang Guibao 已提交
274
分布式稀疏参数服务由paddle serving的Cube模块实现。Cube服务接受的原始数据格式为Hadoop seqfile格式,因此需要对paddle保存出的模型文件进行格式转换。
X
add doc  
xulongteng 已提交
275

W
Wang Guibao 已提交
276
在trainer镜像中,将模型参数转换为seqfile的主要流程是:
X
add doc  
xulongteng 已提交
277

W
Wang Guibao 已提交
278 279
1. 监视训练脚本所在目录的models文件夹,当发现有子目录`pass-1000`时,表示训练任务完成 (默认训练轮次为1000)
2. 调用dumper.py,将models/pass-1000/SparseFeatFactors文件转换成seqfile格式,同时生成一个用于让下游cube-transfer下载完整数据的donefile文件,整个目录结构放到HTTP服务目录下,供下游cube-transfer监听进程检测和下载 (详见本文“大规模稀疏参数服务Cube的部署和使用”一节)
X
add doc  
xulongteng 已提交
280

X
xulongteng 已提交
281 282 283 284
## <span id="head15"> 大规模稀疏参数服务Cube的部署和使用</span>

Cube大规模稀疏参数服务服务组件,用于承载超大规模稀疏参数的查询、更新等各功能。上述分布式训练产出的稀疏参数,在k8s中以http文件服务的形式提供下载;cube则负责将稀疏参数读取、加工,切分成多个分片,灌入稀疏参数服务集群,提供对外访问。

W
Wang Guibao 已提交
285
Cube一共拆分成四个组件,共同完成上述工作:
X
xulongteng 已提交
286

W
Wang Guibao 已提交
287 288 289 290
1. cube-transfer 负责监听上游数据产出,当判断到数据更新时,将数据下载到cube-builder建库端,然后将建库的数据配送到由多个物理节点组成的稀疏参数服务集群
2. cube-builder 负责从上游数据构建cube内部索引格式,并切分成多个分片,完成建库工作
3. cube-server 每个单独的cube服务承载一个分片的cube数据
4. cube-agent 与cube-server伴生部署,负责接受cube-transfer下发的指令,在本地执行实际的数据下载维护等操作
X
xulongteng 已提交
291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313

关于Cube的详细说明文档,请参考[Cube设计文档](https://github.com/PaddlePaddle/Serving/tree/develop/cube/doc/DESIGN.md)。本文仅描述从头部署Cube服务的流程。

### <span id="head16">1. 编译</span>

Cube是Paddle Serving内置的组件,只要按常规步骤编译Serving即可。要注意的是,编译Cube需要Go语言编译器。

```bash
$ git clone https://github.com/PaddlePaddle/Serving.git
$ cd Serving
$ makedir build
$ cd build
$ cmake -DWITH_GPU=OFF .. # 不需要GPU
$ make -jN                # 这里可修改并发编译线程数
$ make install
$ cd output/
$ ls bin
cube  cube-builder  cube-transfer  pdcodegen
$ ls conf
gflags.conf  transfer.conf
```

其中:
W
Wang Guibao 已提交
314 315 316
1) bin/cube, bin/cube-agent, bin/cube-builder, bin/cube-transfer是上述3个组件的可执行文件。**bin/cube是cube-server的可执行文件**
2) conf/gflags.conf是配合bin/cube使用的配置文件,主要包括端口配置等等
3) conf/transfer.conf是配合bin/cube-transfer使用的配置文件,主要包括要监听的上游数据地址等等
X
xulongteng 已提交
317

W
Wang Guibao 已提交
318
接下来我们按cube server/agent, cube-builder, cube-transfer的顺序,介绍Cube的完整部署流程
X
xulongteng 已提交
319 320 321



W
Wang Guibao 已提交
322
### <span id="head17">2. 分片cube server/agent部署</span>
X
xulongteng 已提交
323 324 325 326


#### <span id="head18">2.1 配置文件修改</span>

W
Wang Guibao 已提交
327
首先修改cube server的配置文件,将port改为我们需要的端口号,(当本机内存资源紧张时,将in_mem修改为false将以磁盘访问的模式启动cube server):
X
xulongteng 已提交
328 329 330 331 332 333 334 335 336

```
--port=8000
--dict_split=1
--in_mem=true
```

#### <span id="head19">2.2 拷贝可执行文件和配置文件到物理机</span>

W
Wang Guibao 已提交
337
将bin/cube,bin/cube-agent和conf/gflags.conf拷贝到多个物理机上。假设拷贝好的文件结构如下:
X
xulongteng 已提交
338

X
fix doc  
xulongteng 已提交
339
```bash
X
xulongteng 已提交
340 341 342 343
$ tree
.
|-- bin
|   `-- cube
W
Wang Guibao 已提交
344 345 346
|   `-- cube-agent
|-- conf
|   `-- gflags.conf
X
xulongteng 已提交
347 348
```

W
Wang Guibao 已提交
349
#### <span id="head20">2.3 启动 cube server/agent</span>
X
xulongteng 已提交
350 351 352

```bash
nohup bin/cube &
W
Wang Guibao 已提交
353
nohup bin/cube-agent -P 8001 &
X
xulongteng 已提交
354
```
W
Wang Guibao 已提交
355
其中cube-agent在启动命令中使用 -P 参数指定监听端口号
X
xulongteng 已提交
356 357 358 359 360 361 362

### <span id="head21">3. cube-builder部署</span>

#### <span id="head22">3.1 配置文件修改</span>

cube-builder配置项说明:

W
Wang Guibao 已提交
363
均在启动参数中提交
X
xulongteng 已提交
364

W
Wang Guibao 已提交
365
参数项如下:
X
xulongteng 已提交
366 367

```
W
Wang Guibao 已提交
368 369 370 371 372 373 374 375 376 377 378 379 380
open_builder: Usage : ./open_build --help

Flags from /home/work/dangyifei/open-builder/src/main.cpp:
-cur_version (current version, no need) type: int32 default: 0                //单机builder模式下不需要
-depend_version (depend version, job mode delta need) type: int32 default: 0  //单机builder base模式下不需要,patch模式找到meta_info里的base的key
-dict_name (dict name, no need) type: string default: ""                      //词典名,单机builder模式下不加默认空,用来和版本拼接生成索引文件名
-input_path (source data input path) type: string default: ""               //必须,源数据所在路径,仅支持本地路径
-job_mode (job mode base/delta default:base) type: string default: "base"    //默认base模式,delta模式:-job_mode=delta
-last_version (last version, job mode delta need) type: int32 default: 0     //单机builder base模式下不需要,patch模式找到meta_info里的base的id
-master_address (master address, no need) type: string default: ""           //单机builder模式不需要,会把index meta信息写到本地output/meta_info目录下
-only_build (wheather build need transfer) type: bool default: true         //单机builder模式不需要,代表是不是单机builder,如果false会向master_address发送请求,将index meta信息写到远程
-output_path (source data input path) type: string default: ""       //必须,索引建库数据的输出路径,仅支持本地路径
-shard_num (shard num) type: int32 default: -1             //必须,数据切分的分片数量
X
xulongteng 已提交
381

W
Wang Guibao 已提交
382
```
X
xulongteng 已提交
383
#### <span id="head23">3.2 拷贝可执行文件到物理机</span>
D
Dang Yifei 已提交
384 385 386
需要将bin/cube-builder拷贝到物理机上。  
只利用builder工具建立索引无特殊位置要求,如果接入配送环节使用必须和cube-transfer同机部署。  
假设单独使用builder工具,文件结构如下:  
X
xulongteng 已提交
387

X
fix doc  
xulongteng 已提交
388
```bash
W
Wang Guibao 已提交
389 390 391 392 393
$ tree
`-- cube-builder
|-- source
|   `-- test_seqfile
|-- output
X
xulongteng 已提交
394 395 396
```

#### <span id="head24">3.3 启动cube-builder</span>
W
Wang Guibao 已提交
397
##### 3.3.1接入配送流程
X
fix doc  
xulongteng 已提交
398 399 400 401 402 403 404 405 406 407 408 409 410
拷贝bin/cube-builder和cube-transfer程序到同一机器。  
相关参数已经封装好,只需要在cube-transfer的conf/transfer.conf里配置好cube-builder的地址、源数据和建库数据output的地址即可,  执行cube-transfer时会通过配置文件中的路径调用cube-builder,所以通常不需要手动执行cube-builder。

##### 3.3.2单机builder

**假设分片数为2,词典名为test**

###### base模式  

启动cube-builder命令,参数中的路径需要为绝对路径

```bash
./cube-builder -input_path=${source} -output_path=${output} -shard_num=2 -dict_name=test
W
Wang Guibao 已提交
411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437
```
运行后会根据当前时间戳自动生成建库索引文件夹1565323045_1565323045和meta信息文件夹meta_info结构如下:
```
$ tree
`-- cube-builder
|-- source
|   `-- test_seqfile
`-- output
    |-- 1565323045_1565323045
    |   |-- test_part0
    |   |   |-- data.0
    |   |   |-- data.n
    |   |   |-- index.0
    |   |   `-- index.n
    |   |-- test_part0.tar
    |   |-- test_part0.tar.md5
    |   |-- test_part1
    |   |   |-- data.0
    |   |   |-- data.n
    |   |   |-- index.0
    |   |   `-- index.n
    |   |-- test_part1.tar
    |   `-- test_part1.tar.md5
    `-- meta_info
        |-- 1565323045_1565323045_0_0.json
        `-- 1565323045_1565323045_1_0.json
```
D
Dang Yifei 已提交
438
test_part0.tar和test_part0.tar.md5是shard0分片的数据和md5校验,1565323045_1565323045_0_0.json是0号分片的索引长度和数量,在对应版本的delta建库中需要。  
X
fix doc  
xulongteng 已提交
439
###### delta模式
D
Dang Yifei 已提交
440
需要依赖于上次的base或者delta的id和key,1565323045_1565323045_0_0.json前一个时间戳是id,后一个是key(和分片数据的目录key_id相反),对应cube-builder输入参数-last_version和-depend_version,保持output和dict_name不变(builder会寻找上一轮的index meta信息)。  
X
fix doc  
xulongteng 已提交
441 442 443 444 445

启动cube-builder命令,参数中的路径需要为绝对路径

```bash
./cube-builder -input_path=${source} -output_path=${output} -shard_num=2 -depend_version=1565323045 -last_version=1565323045 -job_mode=delta -dict_name=test
W
Wang Guibao 已提交
446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490
```
运行后会根据当前时间戳自动生成delta建库索引文件夹1565323045_1565326078和meta信息文件夹meta_info结构如下:
```
$ tree
`-- cube-builder
|-- source
|   `-- test_seqfile
`-- output
    |-- 1565323045_1565323045
    |   |-- test_part0
    |   |   |-- data.0
    |   |   |-- data.n
    |   |   |-- index.0
    |   |   `-- index.n
    |   |-- test_part0.tar
    |   |-- test_part0.tar.md5
    |   |-- test_part1
    |   |   |-- data.0
    |   |   |-- data.n
    |   |   |-- index.0
    |   |   `-- index.n
    |   |-- test_part1.tar
    |   `-- test_part1.tar.md5
    |-- 1565323045_1565326078
    |   |-- test_part0
    |   |-- data.0
    |   |   |-- data.n
    |   |   |-- index.0
    |   |   `-- index.n
    |   |-- test_part0.tar
    |   |-- test_part0.tar.md5
    |   |-- test_part1
    |   |   |-- data.0
    |   |   |-- data.n
    |   |   |-- index.0
    |   |   `-- index.n
    |   |-- test_part1.tar
    |   `-- test_part1.tar.md5
    `-- meta_info
        |-- 1565323045_1565323045_0_0.json
        |-- 1565323045_1565323045_0_0.json
        |-- 1565326078_1565323045_0_0.json
        `-- 1565326078_1565323045_1_0.json
```
#### <span id="head241">3.4 seqfile工具</span>
X
fix doc  
xulongteng 已提交
491
builder输入数据的源格式必须为seqfile,key为uint64(输入必须为二进制8个字节),value为序列化的二进制。   
D
Dang Yifei 已提交
492
提供明文转seqfile工具和读seqfile工具,位置在output/tool里kvtool.py和kv_to_seqfile.py。  
W
Wang Guibao 已提交
493 494 495 496 497
kvtool.py 是读seqfile工具,会输出读到的kv信息,参数是文件地址假设在/home/work/test下的seqfile,运行方式如下:
```
python kvtool.py /home/work/test/seqfile
```
kv_to_seqfile.py是明文转seqfile工具,依赖于kvtool.py,会将明文kv转为seqfile文件存储,并输出donefile,在kv_to_seqfile.py的27和30行修改输入和donefile路径:
X
xulongteng 已提交
498 499
```

W
Wang Guibao 已提交
500 501 502 503 504 505 506 507 508 509 510 511 512
BASE_DONEFILE = DATA_PATH + "donefile/base.txt"  #base donefile文件地址
SOURCE_FILE = './source/file.txt' #明文源数据路径
```
要求明文txt内的格式,每行一对kv,用:分割,示例如下:
```
1:1
2:2
10:10 11 12
11:this is eleven
12:value can string
1676869128226002114:48241    37064        91    -539    114    51    -122    269    229    -134    -282
1657749292782759014:167    40        98    27    117    10    -29    15    74    67    -54
```
X
xulongteng 已提交
513 514 515 516
### <span id="head25">4. cube-transfer部署</span>

#### <span id="head26">4.1 cube-transfer配置修改</span>

X
fix doc  
xulongteng 已提交
517
cube-transfer配置文件是conf/transfer.conf,配置比较复杂,配置文件中的路径需要为绝对路径,各个配置项含义如下:
X
xulongteng 已提交
518
```
W
Wang Guibao 已提交
519 520 521 522 523 524 525 526
[default]
dict_name: test_dict                                    //词典名
mode: base_delta                                    //配送模式base_only/base_delta
storage_place: LOCAL                                    //默认LOCAL,表示使用单机builder工具
buildtool_local: /home/work/test-builder/build/cube-builder            //build工具位置,必须在本地,绝对路径
donefile_address: /home/work/test-transfer/test_data/donefile            //donefile位置支持本地路径和远程ftp或者http服务(ftp://或者http://),只到最后文件夹,文件夹内最多2个文件base.txt patch.txt
output_address: /home/work/test-transfer/test_data/output            //build后数据索引输出位置
tmp_address: /home/work/test-transfer/test_data/tmp                //transfer工具运行中临时文件存放位置
X
xulongteng 已提交
527 528
shard_num: 2                                        //分片数
copy_num: 1                                        //每片副本数
W
Wang Guibao 已提交
529 530 531 532 533
deploy_path: /home/work/test_dict                      //不用修改                          
transfer_address: 10.10.10.5                             //cube-transfer本机的ip

[cube_agent]
agent0_0: 10.10.220.15:8001                        //0号分片0号副本的agent ip:port
X
fix doc  
xulongteng 已提交
534
cube0_0: 10.10.220.15:8000:/ssd2/cube_open                //0号分片0号副本的cube,该路径下会存放配送的数据 ip:port:deploy_path
X
xulongteng 已提交
535 536
agent1_0: 10.10.180.40:8001                        //0号分片1号副本的agent ip:port
cube1_0: 10.10.180.40:8000:/home/disk1/cube_open             //0号分片1号副本的cube ,该路径下会存放配送的数据 ip:port:deploy_path
X
xulongteng 已提交
537 538 539 540
```

#### <span id="head27">4.2 拷贝cube-transfer到物理机</span>

W
Wang Guibao 已提交
541
将bin/cube-transfer和conf/transfer.conf拷贝到多个物理机上,构建output和tmp文件夹用来存放配送的中间文件。  
D
Dang Yifei 已提交
542
假设拷贝好的文件结构如下:
X
fix doc  
xulongteng 已提交
543

X
xulongteng 已提交
544
```
W
Wang Guibao 已提交
545 546 547 548 549 550 551
$ tree
.
|-- cube-transfer
|-- output
|-- tmp
`-- conf
    |-- transfer.conf
X
xulongteng 已提交
552 553
```
#### <span id="head28">4.3 启动cube-transfer</span>
X
fix doc  
xulongteng 已提交
554 555
假设启动服务端口8099,-l参数是log等级 --config是配置文件位置,./log文件夹下可以查看cube-transfer的日志
```bash
W
Wang Guibao 已提交
556
./cube-transfer -p 8099 -l 4 --config conf/transfer.conf
X
xulongteng 已提交
557
```
D
Dang Yifei 已提交
558
#### <span id="head281">4.4 cube-transfer支持查询接口</span>
W
Wang Guibao 已提交
559
> 获取当前词典状态  
X
fix doc  
xulongteng 已提交
560
> http://10.10.10.5:8099/dict/info  
W
Wang Guibao 已提交
561 562

> 获取实例当前状态  
X
fix doc  
xulongteng 已提交
563
> http://10.10.10.5:8099/instance/status  
W
Wang Guibao 已提交
564 565

> 获取配送历史从最近的base到当前正在配送的delta  
X
fix doc  
xulongteng 已提交
566
> http://10.10.10.5:8099/dict/deploy/history 
W
Wang Guibao 已提交
567

D
Dang Yifei 已提交
568
#### <span id="head29">4.5 donefile格式协议</span>
W
Wang Guibao 已提交
569

D
Dang Yifei 已提交
570 571
一旦cube-transfer部署完成,它就不断监听我们配置好的donefile数据位置,发现有数据更新后,即启动数据下载,然后通知cube-builder执行建库和配送流程,将新数据配送给各个分片的cube-server。  
id最好使用版本产出时间戳,base和patch每产出一条直接在donefile文件最后加一行即可,文件名固定base.txt、patch.txt  
X
fix doc  
xulongteng 已提交
572

W
Wang Guibao 已提交
573 574 575 576 577 578 579 580 581
>base.txt每行一条,id和key相同,目录下可有多个文件,不能有文件夹
>```
>{"id":"1562000400","key":"1562000400","input":"/home/work/test_data/input/seqfile"}
>```
>patch.txt每行一条,key为base的id
>```
>{"id":"1562000401","key":"1562000400","input":"/home/work/test_data/input/seqfile"}
>{"id":"1562000402","key":"1562000400","input":"/home/work/test_data/input/seqfile"}
>```
X
xulongteng 已提交
582 583 584 585

## <span id="head30"> 预测服务部署</span>

### <span id="head31"> 1、Server端</span>
X
add doc  
xulongteng 已提交
586 587 588 589

通过wget命令从集群获取dense部分模型用于Server端。

```bash
W
Wang Guibao 已提交
590
wget "http://${HTTP_SERVICE_IP}:${HTTP_SERVICE_PORT}/path/to/models"
X
add doc  
xulongteng 已提交
591 592 593 594 595 596
```

K8s集群上CTR预估任务训练完成后,模型参数分成2部分:一是embedding数据,经过dumper.py已经转成hadoop SequenceFile格式,传输给cube建库流程构建索引和灌cube;二是除embedding之外的参数文件,连同save_program.py裁剪后的program,一起配合传输给Serving加载。save_program.py裁剪原始模型的具体背景和详细步骤请参考文档[Paddle Serving CTR预估模型说明](https://github.com/PaddlePaddle/Serving/blob/develop/doc/CTR_PREDICTION.md)

本文介绍Serving使用上述模型参数和program加载模型提供预测服务的流程。

X
xulongteng 已提交
597
#### <span id="head32">1.1 Cube服务</span>
X
add doc  
xulongteng 已提交
598 599 600 601 602

假设Cube服务已经成功部署,用于cube客户端API的配置文件如下所示:

```json
[{
X
xulongteng 已提交
603 604 605 606 607 608 609 610 611 612 613 614 615
"dict_name": "dict",
"shard": 2,
"dup": 1,
"timeout": 200,
"retry": 3,
"backup_request": 100,
"type": "ipport_list",
"load_balancer": "rr",
"nodes": [{
"ipport_list": "list://192.168.1.1:8000"
},{
"ipport_list": "list://192.168.1.2:8000"
}]
X
add doc  
xulongteng 已提交
616 617 618 619 620
}]
```

上述例子中,cube提供外部访问的表名是`dict`,有2个物理分片,分别在192.168.1.1:8000和192.168.1.2:8000

X
xulongteng 已提交
621
#### <span id="head33">1.2 Serving编译</span>
X
add doc  
xulongteng 已提交
622 623 624 625 626 627 628 629 630 631 632 633

截至写本文时,Serving develop分支已经提供了CTR预估服务相关OP,参考[ctr_prediction_op.cpp](https://github.com/PaddlePaddle/Serving/blob/develop/demo-serving/op/ctr_prediction_op.cpp),该OP从client端接收请求后会将每个请求的26个sparse feature id发给cube服务,获得对应的embedding向量,然后填充到模型feed variable对应的LoDTensor,执行预测计算。只要按常规步骤编译Serving即可。

```bash
$ git clone https://github.com/PaddlePaddle/Serving.git
$ cd Serving
$ makedir build
$ cd build
$ cmake -DWITH_GPU=OFF .. # 不需要GPU
$ make -jN                # 这里可修改并发编译线程数
$ make install
$ cd output/demo/serving
W
Wang Guibao 已提交
634 635
$ ls
bin  conf  data  kvdb  log
X
add doc  
xulongteng 已提交
636 637
```

X
xulongteng 已提交
638
#### <span id="head34">1.3 配置修改</span>
X
add doc  
xulongteng 已提交
639

X
xulongteng 已提交
640
##### <span id="head35">1.3.1 conf/gflags.conf</span>
X
add doc  
xulongteng 已提交
641 642 643 644 645 646 647

将--enable_cube改为true:

```json
--enable_cube=true
```

X
xulongteng 已提交
648
##### <span id="head36">1.3.2 conf/model_toolkit.prototxt</span>
X
add doc  
xulongteng 已提交
649 650 651 652 653

Paddle Serving自带的model_toolkit.prototxt如下所示,如有必要可只保留ctr_prediction一个:

```
engines {
X
xulongteng 已提交
654 655 656 657 658 659 660 661
name: "image_classification_resnet"
type: "FLUID_CPU_NATIVE_DIR"
reloadable_meta: "./data/model/paddle/fluid_time_file"
reloadable_type: "timestamp_ne"
model_data_path: "./data/model/paddle/fluid/SE_ResNeXt50_32x4d"
runtime_thread_num: 0
batch_infer_size: 0
enable_batch_align: 0
X
add doc  
xulongteng 已提交
662 663
}
engines {
X
xulongteng 已提交
664 665 666 667 668 669 670 671
name: "text_classification_bow"
type: "FLUID_CPU_ANALYSIS_DIR"
reloadable_meta: "./data/model/paddle/fluid_time_file"
reloadable_type: "timestamp_ne"
model_data_path: "./data/model/paddle/fluid/text_classification_lstm"
runtime_thread_num: 0
batch_infer_size: 0
enable_batch_align: 0
X
add doc  
xulongteng 已提交
672 673 674
}

engines {
X
xulongteng 已提交
675 676 677 678 679 680 681 682 683 684
name: "ctr_prediction"
type: "FLUID_CPU_ANALYSIS_DIR"
reloadable_meta: "./data/model/paddle/fluid_time_file"
reloadable_type: "timestamp_ne"
model_data_path: "./data/model/paddle/fluid/ctr_prediction"
runtime_thread_num: 0
batch_infer_size: 0
enable_batch_align: 0
sparse_param_service_type: REMOTE
sparse_param_service_table_name: "dict"
X
add doc  
xulongteng 已提交
685 686 687 688 689 690
}
```

注意ctr_prediction model有如下2行配置:

```json
X
xulongteng 已提交
691 692
sparse_param_service_type: REMOTE
sparse_param_service_table_name: "dict"
X
add doc  
xulongteng 已提交
693 694
```

X
xulongteng 已提交
695
##### <span id="head37">1.3.3 conf/cube.conf</span>
X
add doc  
xulongteng 已提交
696 697 698 699 700

conf/cube.conf是一个完整的cube配置文件模板,其中只要修改nodes列表为真实的物理节点IP:port列表即可。例如 (与第1节cube配置文件内容一致):

```json
[{
X
xulongteng 已提交
701 702 703 704 705 706 707 708 709 710 711 712 713
"dict_name": "dict",
"shard": 2,
"dup": 1,
"timeout": 200,
"retry": 3,
"backup_request": 100,
"type": "ipport_list",
"load_balancer": "rr",
"nodes": [{
"ipport_list": "list://192.168.1.1:8000"
},{
"ipport_list": "list://192.168.1.2:8000"
}]
X
add doc  
xulongteng 已提交
714 715 716 717 718
}]
```

**注意事项:**如果修改了`dict_name`,需要同步修改1.3.2节中`sparse_param_service_table_name`字段

X
xulongteng 已提交
719
##### <span id="head38">1.3.4 模型文件</span>
X
add doc  
xulongteng 已提交
720

W
Wang Guibao 已提交
721 722 723
Paddle Serving自带了一个可以工作的CTR预估模型,是从BCE上下载下来的,其制作方法为:
1. 分布式训练CTR预估任务,保存模型program和参数文件
2. 用save_program.py保存一份用于预测的program (文件名为**model**)。save_program.py随trainer docker image发布
W
Wang Guibao 已提交
724
3. 第2步中保存的program (**model**) 覆盖到第1步保存的模型文件夹中**model**文件,打包成.tar.gz上传到BCE
X
add doc  
xulongteng 已提交
725 726 727

如果只是为了验证demo流程,serving此时已经可以用自带的CTR模型加载模型并提供预测服务能力。

W
Wang Guibao 已提交
728
为了应用重新训练的模型,只需要从k8s集群暴露的http服务下载新的ctr_model.tar.gz,解压到data/model/paddle/fluid下,并将内容移至原来的ctr_prediction目录即可:
W
Wang Guibao 已提交
729 730
```bash
$ cd data/model/paddle/fluid
W
Wang Guibao 已提交
731
$ wget http://${HTTP_SERVICE_IP}:${HTTP_SERVICE_PORT}/data/ctr_model.tar.gz
W
Wang Guibao 已提交
732
$ tar zxvf ctr_model.tar.gz # 假设解压出一个inference_only目录
W
Wang Guibao 已提交
733
$ rm -rf ctr_prediction     # 删除旧的ctr_prediction目录下内容
W
Wang Guibao 已提交
734
$ cp -r inference_only/* ctr_prediction
W
Wang Guibao 已提交
735 736 737 738 739 740 741
$ cd ../../../../           # 切换至serving所在目录
$ ls
bin  conf  data  kvdb  log
$ killall serving           # 杀死旧的serving进程
$ bin/serving &             # 重启serving
```

W
Wang Guibao 已提交
742
从K8S集群暴露的http服务下载训练模型,请参考文档[PaddlePaddle分布式训练和Serving流程化部署](http://icode.baidu.com/repos/baidu/personal-code/wangguibao/blob/master:ctr-embedding-to-sequencefile/path/to/doc/DISTRIBUTED_TRANING_AND_SERVING.md)
X
add doc  
xulongteng 已提交
743

X
xulongteng 已提交
744
#### <span id="head39">1.4 启动Serving</span>
X
add doc  
xulongteng 已提交
745 746 747

执行`./bin/serving  `启动serving服务,在./log路径下可以查看serving日志。

X
xulongteng 已提交
748
### <span id="head40"> 2、Client端</span>
X
add doc  
xulongteng 已提交
749 750 751 752 753

参考[从零开始写一个预测服务:client端]([https://github.com/PaddlePaddle/Serving/blob/develop/doc/CREATING.md#3-client%E7%AB%AF](https://github.com/PaddlePaddle/Serving/blob/develop/doc/CREATING.md#3-client端))文档,实现client端代码。

文档中使用的CTR预估任务client端代码存放在Serving代码库demo-client路径下,链接[ctr_prediction.cpp](https://github.com/PaddlePaddle/Serving/blob/develop/demo-client/src/ctr_prediction.cpp)

X
xulongteng 已提交
754
#### <span id="head41">2.1 测试数据</span>
X
add doc  
xulongteng 已提交
755 756 757

CTR预估任务样例使用的数据来自于[原始模型](https://github.com/PaddlePaddle/models/tree/develop/PaddleRec/ctr)的测试数据,在样例中提供了1000个测试样本,如果需要更多样本可以参照原始模型下载数据的[脚本](https://github.com/PaddlePaddle/models/blob/develop/PaddleRec/ctr/data/download.sh)

X
xulongteng 已提交
758
#### <span id="head42">2.2 Client编译与部署</span>
X
add doc  
xulongteng 已提交
759

X
fix doc  
xulongteng 已提交
760
按照1.2Serving编译部分完成编译后,client端文件在output/demo/client/ctr_prediction路径下。
X
add doc  
xulongteng 已提交
761

X
xulongteng 已提交
762
##### <span id="head43">2.2.1 配置修改</span>
X
add doc  
xulongteng 已提交
763 764 765 766 767

修改conf/predictors.prototxt文件ctr_prediction_service部分

```
predictors {
X
xulongteng 已提交
768 769 770 771 772 773 774 775 776 777 778 779
name: "ctr_prediction_service"
service_name: "baidu.paddle_serving.predictor.ctr_prediction.CTRPredictionService"
endpoint_router: "WeightedRandomRender"
weighted_random_render_conf {
variant_weight_list: "50"
}
variants {
tag: "var1"
naming_conf {
cluster: "list://127.0.0.1:8010"
}
}
X
add doc  
xulongteng 已提交
780 781 782 783 784
}
```

配置Server端ip与端口号,默认为本机ip、8010端口。

X
xulongteng 已提交
785
##### <span id="head44">2.2.2 运行服务</span>
X
add doc  
xulongteng 已提交
786

X
fix toc  
xulongteng 已提交
787
执行`./bin/ctr_predictoin`启动client端,在./log路径下可以看到client端执行的日志。