mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

【模型】Bert base 

【修改说明】cluner支持SDK、mxbase推理
【验证结果】容器中推理 OK
【修改人】chenshushu
【评审人】
This commit is contained in:
chenshushu 2021-07-20 10:11:01 +08:00
parent 0edb6d0855
commit e2635f0ef6
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
model_zoo/official/nlp/bert/infer/utils/data_processor_seq.py
View File

@ -127,7 +127,7 @@ def get_all_path(output_path):
token_path = os.path.join(output_path, "02_data") token_path = os.path.join(output_path, "02_data")
label_path = os.path.join(output_path, "03_data") label_path = os.path.join(output_path, "03_data")
for path in [ids_path, mask_path, token_path, label_path]: for path in [ids_path, mask_path, token_path, label_path]:
os.makedirs(path, 0o755) os.makedirs(path, 0o755, exist_ok=True)
return ids_path, mask_path, token_path, label_path return ids_path, mask_path, token_path, label_path