Skip to content

Commit 0e9d48e

Browse files
finish WDL and wait for supervisor's checking
1 parent 8c99e4b commit 0e9d48e

File tree

4 files changed

+785
-10
lines changed

4 files changed

+785
-10
lines changed
Lines changed: 52 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,52 @@
1+
## How to use prepare_savedmodel.py to get savedmodel
2+
3+
- Current support model: \
4+
BST, DBMTL, DeepFM, DIEN, DIN, DLRM, DSSM, ESMM, MMoE, SimpleMultiTask, WDL
5+
6+
- Usage: \
7+
For every model listed above, there is a prepare_savedmodel.py. To run this script please firstly ensure you have gotten the checkpoint file from training. To use prepare_savedmodel.py, please use:
8+
9+
```
10+
cd [modelfolder]
11+
python prepare_savedmodel.py --checkpoint [ckpt path]
12+
```
13+
- If you choose --bf16 during training stage, please also add it above. For example:
14+
15+
```
16+
cd [modelfolder]
17+
python prepare_savedmodel.py --bf16 --checkpoint [ckpt path]
18+
```
19+
20+
21+
- Example: \
22+
This is an example for BST model without bf16 feature
23+
```
24+
cd modelzoo/BST
25+
python prepare_savedmodel.py --checkpoint ./result/model_BST_1657777492
26+
```
27+
28+
- Output: \
29+
The savedmodel will be stored under ./savedmodels folder
30+
31+
32+
## How to use start_serving.cc to retrieving serving result
33+
34+
- Functionality: \
35+
start_serving.cc provides functionality such that you can get serving result after getting the savedmodel.
36+
37+
- Parameter:
38+
1. At the start of main(), there is a file_path variable. Please substitude it to your own evaluation file path (the format should be the same as training one)
39+
2. Please edit the savedmodel and checkpoint path in the model_config which is at the very begining.
40+
41+
- Usage:
42+
1. Please edit BUILD file to add the start_serving
43+
2. Please make sure you have installed serving part properly
44+
3. Then go back to the Deeprec folder and do as the follows, assuming the cc_binary name in BUILD is "wdl_demo":
45+
```
46+
bazel build //serving/processor/tests:wdl_demo
47+
bazel-bin/serving/processor/tests/wdl_demo
48+
```
49+
50+
51+
52+

modelzoo/features/EmbeddingVariable/WDL/data/README.md

Lines changed: 0 additions & 10 deletions
This file was deleted.

0 commit comments

Comments
 (0)