Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
BERT-Large-squad_onnxruntime
Commits
b5a7f01c
Commit
b5a7f01c
authored
Oct 13, 2023
by
chenxj
Browse files
update README
parent
1d300d4e
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
4 additions
and
3 deletions
+4
-3
README.md
README.md
+4
-3
No files found.
README.md
View file @
b5a7f01c
...
@@ -16,12 +16,13 @@ bert_large_squad模型的主要参数为:24个transformer层、1024个hidden s
...
@@ -16,12 +16,13 @@ bert_large_squad模型的主要参数为:24个transformer层、1024个hidden s
```
```
docker pull image.sourcefind.cn:5000/dcu/admin/base/custom:ort1.14.0_migraphx3.0.0-dtk22.10.1
docker pull image.sourcefind.cn:5000/dcu/admin/base/custom:ort1.14.0_migraphx3.0.0-dtk22.10.1
docker run -d -t --privileged --device=/dev/kfd --device=/dev/dri/ --network=host --group-add video --name bert-test image.sourcefind.cn:5000/dcu/admin/base/custom:ort1.14.0_migraphx3.0.0-dtk22.10.1
docker run -d -t --privileged --device=/dev/kfd --device=/dev/dri/ --network=host --group-add video --name bert-test image.sourcefind.cn:5000/dcu/admin/base/custom:ort1.14.0_migraphx3.0.0-dtk22.10.1
docker exec -it bert-test bash
```
```
在
[
huggingface
](
https://huggingface.co/ctuning/mlperf-inference-bert-onnx-fp32-squad-v1.1
)
下载模型model.onnx到当前目录
在
[
huggingface
](
https://huggingface.co/ctuning/mlperf-inference-bert-onnx-fp32-squad-v1.1
)
下载模型model.onnx到当前目录
执行fp16转换
执行fp16转换
```
```
pip3 install
onnxmltools
pip3 install
-r requirements.txt
python3 fp16-convert.py
python3 fp16-convert.py
```
```
## 推理
## 推理
...
@@ -30,7 +31,7 @@ python3 main.py
...
@@ -30,7 +31,7 @@ python3 main.py
```
```
## result
## result


###
性能和精度数据
###
精度
暂无
暂无
## 应用场景
## 应用场景
### 算法类别
### 算法类别
...
@@ -39,6 +40,6 @@ nlp
...
@@ -39,6 +40,6 @@ nlp
问答系统
问答系统
## 源码仓库及问题反馈
## 源码仓库及问题反馈
https://developer.hpccube.com/codes/modelzoo/bert_large_squad_onnx
https://developer.hpccube.com/codes/modelzoo/bert_large_squad_onnx
## 参考
## 参考
资料
https://github.com/google-research/bert
https://github.com/google-research/bert
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment