Skip to content

Commit d4cf11a

Browse files
author
tianxin
authored
fix ernie_text_matching ci config (#877)
* fix ernie_text_matching ci config
1 parent 303ef9c commit d4cf11a

File tree

1 file changed

+8
-6
lines changed

1 file changed

+8
-6
lines changed

tests/ernie_text_matching/ernie_text_matching_params.txt

Lines changed: 8 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,16 +1,15 @@
1-
21
===========================train_params===========================
32
model_name:ernie_text_matching
43
python:python
54
gpu_list:0|0,1
6-
Global.use_gpu:null
7-
Global.auto_cast:null
5+
null:null
6+
null:null
87
--epoch:1
98
--save_dir:./output/
109
--batch_size:32
11-
Global.pretrained_model:null
12-
train_model_name:null
13-
train_infer_img_dir:null
10+
null:null
11+
null:null
12+
null:null
1413
null:null
1514
##
1615
trainer:norm
@@ -35,6 +34,9 @@ distill_export:null
3534
null:null
3635
null:null
3736
##
37+
infer_model:null
38+
infer_export:null
39+
infer_quant:null
3840
inference:./ernie_text_matching/predict.py
3941
--device:cpu|gpu
4042
--enable_mkldnn:True|False

0 commit comments

Comments
 (0)