Skip to content

Commit 7336d52

Browse files
authored
Fix windows dtype (#1480)
* fix variable name * fix windows dtype
1 parent fa1fa75 commit 7336d52

File tree

4 files changed

+12
-8
lines changed

4 files changed

+12
-8
lines changed

examples/few_shot/efl/deploy/python/predict.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -163,8 +163,8 @@ def predict(self, data, tokenizer):
163163
examples.append((input_ids, segment_ids))
164164

165165
batchify_fn = lambda samples, fn=Tuple(
166-
Pad(axis=0, pad_val=tokenizer.pad_token_id), # input
167-
Pad(axis=0, pad_val=tokenizer.pad_token_id), # segment
166+
Pad(axis=0, pad_val=tokenizer.pad_token_id, dtype="int64"), # input
167+
Pad(axis=0, pad_val=tokenizer.pad_token_id, dtype="int64"), # segment
168168
): fn(samples)
169169

170170
if args.benchmark:

examples/sentiment_analysis/skep/deploy/python/predict.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -43,8 +43,8 @@ def convert_example(example,
4343
is_test=False):
4444
text = example
4545
encoded_inputs = tokenizer(text=text, max_seq_len=max_seq_length)
46-
input_ids = np.array(tokenized_input['input_ids'], dtype="int64")
47-
token_type_ids = np.array(tokenized_input['token_type_ids'], dtype="int64")
46+
input_ids = np.array(encoded_inputs['input_ids'], dtype="int64")
47+
token_type_ids = np.array(encoded_inputs['token_type_ids'], dtype="int64")
4848
return input_ids, token_type_ids
4949

5050

examples/text_matching/ernie_matching/README.md

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -153,14 +153,18 @@ python -u -m paddle.distributed.launch --gpus "0" \
153153
#### 模型导出
154154
使用动态图训练结束之后,可以使用静态图导出工具 `export_model.py` 将动态图参数导出成静态图参数。 执行如下命令:
155155

156-
`python export_model.py --params_path ernie_ckpt/model_80.pdparams --output_path=./output`
156+
```shell
157+
python export_model.py --params_path checkpoints/model_300/model_state.pdparams --output_path=./output
158+
```
157159

158160
其中`params_path`是指动态图训练保存的参数路径,`output_path`是指静态图参数导出路径。
159161

160162
#### 预测部署
161163
导出静态图模型之后,可以基于静态图模型进行预测,`deploy/python/predict.py` 文件提供了静态图预测示例。执行如下命令:
162164

163-
`python deploy/python/predict.py --model_dir ./output`
165+
```shell
166+
python deploy/python/predict.py --model_dir ./output
167+
```
164168

165169
## Reference
166170
[1] Xin Liu, Qingcai Chen, Chong Deng, Huajun Zeng, Jing Chen, Dongfang Li, Buzhou Tang, LCQMC: A Large-scale Chinese Question Matching Corpus,COLING2018.

examples/text_matching/ernie_matching/deploy/python/predict.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -176,8 +176,8 @@ def predict(self, data, tokenizer, label_map):
176176
examples.append((input_ids, segment_ids))
177177

178178
batchify_fn = lambda samples, fn=Tuple(
179-
Pad(axis=0, pad_val=tokenizer.pad_token_id), # input
180-
Pad(axis=0, pad_val=tokenizer.pad_token_id), # segment
179+
Pad(axis=0, pad_val=tokenizer.pad_token_id, dtype="int64"), # input
180+
Pad(axis=0, pad_val=tokenizer.pad_token_id, dtype="int64"), # segment
181181
): fn(samples)
182182

183183
if args.benchmark:

0 commit comments

Comments
 (0)