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_onnxruntime
Commits
3d102d61
"tests/python/pytorch/nn/test_nn.py" did not exist on "dc5035b1f32091847e512e2f9786e08e05495fa4"
Commit
3d102d61
authored
Oct 16, 2023
by
yangql
Browse files
新增方法二
parent
9ef4576d
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
9 additions
and
1 deletion
+9
-1
README.md
README.md
+9
-1
No files found.
README.md
View file @
3d102d61
...
@@ -18,7 +18,7 @@ Segment Embedding:表明这个词属于哪个句子(NSP需要两个句子)
...
@@ -18,7 +18,7 @@ Segment Embedding:表明这个词属于哪个句子(NSP需要两个句子)
## 环境配置
## 环境配置
### Docker
### Docker
(方法一)
拉取镜像:
拉取镜像:
...
@@ -34,6 +34,14 @@ docker run --shm-size 16g --network=host --name=bert_ort --privileged --device=/
...
@@ -34,6 +34,14 @@ docker run --shm-size 16g --network=host --name=bert_ort --privileged --device=/
# 激活dtk
# 激活dtk
source /opt/dtk/env.sh
source /opt/dtk/env.sh
```
```
### Dockerfile(方法二)
```
cd ./docker
docker build --no-cache -t bert_onnxruntime:2.0 .
docker run --shm-size 16g --network=host --name=bert_ort --privileged --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v $PWD/bert_onnxruntime:/home/bert_onnxruntime -it <Your Image ID> /bin/bash
```
## 数据集
## 数据集
通过对模型提出问题,模型完成问题回答任务。
通过对模型提出问题,模型完成问题回答任务。
...
...
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