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_migraphx
Commits
e37eee5a
Commit
e37eee5a
authored
Oct 24, 2023
by
liucong
Browse files
修改readme
parent
501e81d6
Changes
3
Show whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
18 additions
and
3 deletions
+18
-3
README.md
README.md
+15
-2
docker/Dockerfile
docker/Dockerfile
+2
-0
model.properties
model.properties
+1
-1
No files found.
README.md
View file @
e37eee5a
...
...
@@ -18,7 +18,7 @@ BERT的全称为Bidirectional Encoder Representation from Transformers,是一
## 环境配置
### Docker
### Docker
(方法一)
拉取镜像:
...
...
@@ -35,6 +35,15 @@ docker run --shm-size 16g --network=host --name=bert_migraphx --privileged --dev
source /opt/dtk/env.sh
```
### Dockerfile(方法二)
```
cd ./docker
docker build --no-cache -t bert_migraphx:2.0 .
docker run --shm-size 16g --network=host --name=bert_migraphx --privileged --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v $PWD/bert_migraphx:/home/bert_migraphx -it <Your Image ID> /bin/bash
```
## 数据集
在界面中根据提示输入问题,模型预测出答案。
...
...
@@ -134,6 +143,10 @@ question:What is ROCm built for?
answer:scale
```
### 精度
无
## 应用场景
### 算法类别
...
...
@@ -148,6 +161,6 @@ answer:scale
https://developer.hpccube.com/codes/modelzoo/bert_migraphx
## 参考
## 参考
资料
https://github.com/ROCmSoftwarePlatform/AMDMIGraphX/tree/develop/examples/nlp/python_bert_squad
\ No newline at end of file
docker/Dockerfile
0 → 100644
View file @
e37eee5a
FROM
image.sourcefind.cn:5000/dcu/admin/base/migraphx:4.0.0-centos7.6-dtk23.04.1-py38-latest
RUN
source
/opt/dtk/env.sh
model.properties
View file @
e37eee5a
...
...
@@ -5,6 +5,6 @@ modelName=bert_migraphx
# 模型描述
modelDescription
=
Bert是一种预训练的语言表征模型。
# 应用场景
appScenario
=
推理,NLP,
对话问答
appScenario
=
推理,NLP,
政府,科研,金融,教育
# 框架类型
frameType
=
migraphx
\ No newline at end of file
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